Failsafe if we cannot store as npz revert to a pure pickle file.

Python-Ensemble-Toolbox · Jan 1, 2024 · 6d00071 · 6d00071
1 parent 4996063
commit 6d00071
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pipt/misc_tools/analysis_tools.py b/pipt/misc_tools/analysis_tools.py
@@ -659,7 +659,11 @@ def save_analysisdebug(ind_save, **kwargs):
     is passed to np.savez (kwargs) the variable will be stored with their original name.
     """
     # Save input variables
-    np.savez('debug_analysis_step_{0}'.format(str(ind_save)), **kwargs)
+    try:
+        np.savez('debug_analysis_step_{0}'.format(str(ind_save)), **kwargs)
+    except: # if npz save fails dump to a pickle file
+        with open(f'debug_analysis_step_{ind_save}.p', 'wb') as file:
+            pickle.dump(kwargs, file)
 
 
 def get_list_data_types(obs_data, assim_index):