Merge pull request #298 from MiraGeoscience/GEOPY-2557

GEOPY-2557: Save sensitivities option ignored

Author: domfournier

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

tests/run_tests/driver_mt_test.py

@@ -178,6 +178,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)