Code update

PiperOrigin-RevId: 690821682
google-research · Oct 29, 2024 · 49ded0f · 49ded0f
1 parent b07ffdc
commit 49ded0f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/swirl_dynamics/data/hdf5_utils.py b/swirl_dynamics/data/hdf5_utils.py
@@ -91,5 +91,10 @@ def save_array_dict(save_path: epath.PathLike, data: Mapping[str, Any]) -> None:
   with h5py.File(bio, "w") as f:
     _save_array_dict(f, data)
 
+  # Creates parent directory if it does not exist.
+  save_dir = epath.Path(save_path).parent
+  if not filesys.exists(save_dir):
+    filesys.makedirs(save_dir)
+
   with filesys.gfile.GFile(save_path, "w") as f:
     f.write(bio.getvalue())