numpy dtype classes are strange

SBC-Utrecht · Aug 14, 2024 · 404e520 · 404e520
1 parent fe133b3
commit 404e520
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/pytom_tm/tmjob.py b/src/pytom_tm/tmjob.py
@@ -531,8 +531,8 @@ def write_to_json(self, file_name: pathlib.Path) -> None:
         for key, value in d.items():
             if isinstance(value, pathlib.Path):
                 d[key] = str(value)
-        # hardcode tpe conversion
-        d["output_dtype"] = str(d["output_dtype"])
+        # wrangle dtype conversion
+        d["output_dtype"] = str(np.dtype(d["output_dtype"]))
         with open(file_name, "w") as fstream:
             json.dump(d, fstream, indent=4)