Added try block in load_from_path()

kazewong · Sep 2, 2024 · 7309e44 · 7309e44
1 parent c906c67
commit 7309e44
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/jimgw/single_event/runManager.py b/src/jimgw/single_event/runManager.py
@@ -128,9 +128,12 @@ def load_from_path(self, path: str) -> SingleEventRun:
             and "local_sampler_arg" in data["jim_parameters"]
             and "step_size" in data["jim_parameters"]["local_sampler_arg"]
         ):
-            data["jim_parameters"]["local_sampler_arg"]["step_size"] = jnp.array(
-                data["jim_parameters"]["local_sampler_arg"]["step_size"]
-            )
+            try:
+                data["jim_parameters"]["local_sampler_arg"]["step_size"] = jnp.array(
+                    data["jim_parameters"]["local_sampler_arg"]["step_size"]
+                )
+            except Exception as e:
+                print(f"Error in loading step_size: {e}")
         return SingleEventRun(**data)
 
     ### Initialization functions ###