From e905d060868018b481d62017e6aef3433bd686bc Mon Sep 17 00:00:00 2001 From: dominiquef Date: Thu, 19 Jun 2025 10:52:36 -0700 Subject: [PATCH] Remove setting of workspace --- param_sweeps/driver.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/param_sweeps/driver.py b/param_sweeps/driver.py index 277e5f3..d5f83e5 100644 --- a/param_sweeps/driver.py +++ b/param_sweeps/driver.py @@ -104,12 +104,11 @@ def __init__(self, params: SweepParams): raise ValueError("Workspace must be specified.") self.params: SweepParams = params - self.workspace: Workspace = params.geoh5 - - if isinstance(self.workspace.h5file, BytesIO) or self.workspace.h5file is None: + workspace = params.geoh5 + if isinstance(workspace.h5file, BytesIO) or workspace.h5file is None: raise ValueError("Workspace must be saved to disk.") - self.working_directory = str(Path(self.workspace.h5file).parent) + self.working_directory = str(Path(workspace.h5file).parent) lookup = self.get_lookup() self.write_files(lookup) @@ -153,6 +152,9 @@ def update_lookup(self, lookup: dict, gather_first: bool = False): def write_files(self, lookup): """Write ui.geoh5 and ui.json files for sweep trials.""" + if self.params.worker_uijson is None: + raise ValueError("Worker ui.json file path must be specified.") + ifile = InputFile.read_ui_json(self.params.worker_uijson) with ifile.data["geoh5"].open(mode="r") as workspace: for name, trial in lookup.items():