Save version info with Solution (#72)

loganbvh · Feb 6, 2024 · 4926f81 · 4926f81
1 parent 0a2e0be
commit 4926f81
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tdgl/solution/solution.py b/tdgl/solution/solution.py
@@ -891,6 +891,7 @@ def serialize_func(func, name, h5group):
             save_context = nullcontext(h5file)
 
         with save_context as f:
+            f.require_group("version_info").attrs.update(self.version_info)
             if "mesh" in f:
                 del f["mesh"]
             data_grp = f.require_group("data")
@@ -974,9 +975,7 @@ def deserialize_func(name, h5group):
 
         with h5py.File(path, "r") as f:
             grp = f["solution"]
-            options_kwargs = dict()
-            for k, v in grp["options"].attrs.items():
-                options_kwargs[k] = v
+            options_kwargs = dict(grp["options"].attrs)
             options = SolverOptions(**options_kwargs)
             options.validate()
             time_created = datetime.fromisoformat(grp.attrs["time_created"])