BUG: Fix loading scalars from npz file

rochefort-lab · Jul 10, 2021 · 65709ea · 65709ea
1 parent 4441bcd
commit 65709ea
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/fissa/core.py b/fissa/core.py
@@ -685,9 +685,9 @@ def load(self, path=None):
             value = cache[field]
             if np.array_equal(value, None):
                 value = None
-            elif value.size == 1 and str(value) == value:
-                # Handle loading string parameters
-                value = str(value)
+            elif value.ndim == 0:
+                # Handle loading scalars
+                value = value.item()
             setattr(self, field, value)
 
     def separation_prep(self, redo=False):