make sure json can handle numpy arrays

Author: nbbb24

ecg_bench/utils/dir_file_utils.py

@@ -21,8 +21,18 @@ def open_json(path: Union[str, Path]) -> dict:
     @staticmethod
     def save_json(data: dict, path: Union[str, Path]):
         """Save a dictionary to a JSON file."""
+        def convert_numpy(obj):
+            if isinstance(obj, np.ndarray):
+                return obj.tolist()
+            elif isinstance(obj, dict):
+                return {key: convert_numpy(value) for key, value in obj.items()}
+            elif isinstance(obj, list):
+                return [convert_numpy(item) for item in obj]
+            else:
+                return obj
+        
         with open(path, 'w') as f:
-            json.dump(data, f)
+            json.dump(convert_numpy(data), f)
 
     @staticmethod
     def get_system_prompt(system_prompt_path: Union[str, Path]):