diff --git a/simpeg_drivers/components/factories/directives_factory.py b/simpeg_drivers/components/factories/directives_factory.py index ad5a7296..04250362 100644 --- a/simpeg_drivers/components/factories/directives_factory.py +++ b/simpeg_drivers/components/factories/directives_factory.py @@ -144,7 +144,7 @@ def save_directives(self): ]: save_directive = getattr(self, directive) if save_directive is not None: - directives_list.append(getattr(self, directive)) + directives_list.append(save_directive) if ( isinstance(save_directive, directives.SaveDataGeoH5) @@ -200,8 +200,10 @@ def save_property_group(self): @property def save_sensitivities_directive(self): """""" - if self._save_sensitivities_directive is None and isinstance( - self.params, BaseInversionOptions + if ( + self._save_sensitivities_directive is None + and isinstance(self.params, BaseInversionOptions) + and self.params.directives.save_sensitivities ): self._save_sensitivities_directive = SaveSensitivitiesGeoh5Factory( self.params diff --git a/tests/run_tests/driver_mt_test.py b/tests/run_tests/driver_mt_test.py index 9bcc4a00..ad4a90c4 100644 --- a/tests/run_tests/driver_mt_test.py +++ b/tests/run_tests/driver_mt_test.py @@ -185,6 +185,7 @@ def test_magnetotellurics_run(tmp_path: Path, max_iterations=1, pytest=True): driver.params.geoh5.h5file, driver.params.out_group.uid ) output["data"] = orig_zyy_real_1 + assert not run_ws.get_entity("Iteration_0_sensitivities")[0] if pytest: check_target(output, target_run, tolerance=0.2) nan_ind = np.isnan(run_ws.get_entity("Iteration_0_model")[0].values)