add unit test call (otherwise it doesn't run on my computer) and repl…

…ace None by NoneType
pyiron · Jan 24, 2025 · 1f08025 · 1f08025
1 parent eb00b9e
commit 1f08025
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/pyiron_workflow/mixin/preview.py b/pyiron_workflow/mixin/preview.py
@@ -207,6 +207,9 @@ def _get_type_hints(cls) -> dict:
         type_dict = {
             k: v for k, v in type_dict.items() if v != inspect.Parameter.empty
         }
+        for key, value in type_dict.items():
+            if value is None:
+                type_dict[key] = type(None)
         return type_dict
 
     @classmethod

diff --git a/tests/unit/mixin/test_preview.py b/tests/unit/mixin/test_preview.py
@@ -205,3 +205,7 @@ def __source_code_not_available(x):
                 log.output,
                 msg="Verify that the expected warning appears in the log",
             )
+
+
+if __name__ == "__main__":
+    unittest.main()