From 4a1396e771911272b3c49fa8f8196349fb6ae533 Mon Sep 17 00:00:00 2001 From: domfournier Date: Thu, 10 Jul 2025 10:17:01 -0700 Subject: [PATCH] Add flags for opening and closing the file --- simpeg/directives/_save_geoh5.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/simpeg/directives/_save_geoh5.py b/simpeg/directives/_save_geoh5.py index 4cd3cb8aa2..bc289b3f46 100644 --- a/simpeg/directives/_save_geoh5.py +++ b/simpeg/directives/_save_geoh5.py @@ -41,17 +41,22 @@ class BaseSaveGeoH5(InversionDirective, ABC): def __init__( self, h5_object, + *, dmisfit=None, label: str | None = None, channels: list[str] = ("",), components: list[str] = ("",), association: str | None = None, + open_geoh5: bool = False, + close_geoh5: bool = False, **kwargs, ): self.label = label self.channels = channels self.components = components self.h5_object = h5_object + self.open_geoh5 = open_geoh5 + self.close_geoh5 = close_geoh5 if association is not None: self.association = association @@ -61,10 +66,20 @@ def __init__( ) def initialize(self): + if self.open_geoh5: + self._geoh5.open(mode="r+") + self.write(0) + if self.close_geoh5: + self._geoh5.close() + def endIter(self): + if self.open_geoh5: + self._geoh5.open(mode="r+") self.write(self.opt.iter) + if self.close_geoh5: + self._geoh5.close() def get_names( self, component: str, channel: str, iteration: int