From 61a54aed90a65c780d38f9275d62bed34a94def6 Mon Sep 17 00:00:00 2001 From: benjamink Date: Mon, 1 Dec 2025 14:02:32 -0800 Subject: [PATCH 1/2] Use existing SimPEGGroup if it exists --- simpeg_drivers/line_sweep/driver.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/simpeg_drivers/line_sweep/driver.py b/simpeg_drivers/line_sweep/driver.py index 1e5a7f84..9159ff9a 100644 --- a/simpeg_drivers/line_sweep/driver.py +++ b/simpeg_drivers/line_sweep/driver.py @@ -69,9 +69,13 @@ def validate_out_group(self, out_group: SimPEGGroup | None) -> SimPEGGroup: return out_group with fetch_active_workspace(self.workspace, mode="r+"): - out_group = SimPEGGroup.create( - self.workspace, name=self.batch2d_params.title - ) + out_groups = self.workspace.get_entity(self.batch2d_params.title) + if out_groups: + out_group = out_groups[0] + else: + out_group = SimPEGGroup.create( + self.workspace, name=self.batch2d_params.title + ) return out_group From f3356dc7354cd8248db2821be555f2d628b48e5c Mon Sep 17 00:00:00 2001 From: benjamink Date: Mon, 1 Dec 2025 14:36:07 -0800 Subject: [PATCH 2/2] check for none instead of empty list --- simpeg_drivers/line_sweep/driver.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/simpeg_drivers/line_sweep/driver.py b/simpeg_drivers/line_sweep/driver.py index 9159ff9a..b726e90a 100644 --- a/simpeg_drivers/line_sweep/driver.py +++ b/simpeg_drivers/line_sweep/driver.py @@ -69,10 +69,8 @@ def validate_out_group(self, out_group: SimPEGGroup | None) -> SimPEGGroup: return out_group with fetch_active_workspace(self.workspace, mode="r+"): - out_groups = self.workspace.get_entity(self.batch2d_params.title) - if out_groups: - out_group = out_groups[0] - else: + out_group = self.workspace.get_entity(self.batch2d_params.title)[0] + if out_group is None: out_group = SimPEGGroup.create( self.workspace, name=self.batch2d_params.title )