fix-226

punch-mission · Nov 16, 2024 · ac90998 · ac90998
1 parent f4e0755
commit ac90998
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/regularizepsf/transform.py b/regularizepsf/transform.py
@@ -173,6 +173,7 @@ def save(self, path: pathlib.Path) -> None:
         None
 
         """
+        path = pathlib.Path(path)
         if path.suffix == ".h5":
             with h5py.File(path, "w") as f:
                 f.create_dataset("coordinates", data=self.coordinates)
@@ -201,6 +202,7 @@ def load(cls, path: pathlib.Path) -> ArrayPSFTransform:
         PSFTransform
 
         """
+        path = pathlib.Path(path)
         if path.suffix == ".h5":
             with h5py.File(path, "r") as f:
                 coordinates = [tuple(c) for c in f["coordinates"][:]]