diff --git a/simpeg_drivers/joint/joint_petrophysics/driver.py b/simpeg_drivers/joint/joint_petrophysics/driver.py index 5e9bacf8..0b857294 100644 --- a/simpeg_drivers/joint/joint_petrophysics/driver.py +++ b/simpeg_drivers/joint/joint_petrophysics/driver.py @@ -59,23 +59,25 @@ def directives(self): ) ) directives_list += self._get_global_model_save_directives() - directives_list.append( - directives.SavePGIModel( - self.inversion_mesh.entity, - self.pgi_regularization, - self.geo_units, - [driver.params.physical_property for driver in self.drivers], - transforms=[ - lambda x: np.r_[list(self.geo_units)][self.class_mapping][ - x.astype(int) - ], - maps.InjectActiveCells( - self.inversion_mesh.mesh, self.models.active_cells, 0 - ), - ], - reference_type=self.params.models.petrophysical_model.entity_type, - ) - ) + + # TODO: To bring back once we let the classification change + # directives_list.append( + # directives.SavePGIModel( + # self.inversion_mesh.entity, + # self.pgi_regularization, + # self.geo_units, + # [driver.params.physical_property for driver in self.drivers], + # transforms=[ + # lambda x: np.r_[list(self.geo_units)][self.class_mapping][ + # x.astype(int) + # ], + # maps.InjectActiveCells( + # self.inversion_mesh.mesh, self.models.active_cells, 0 + # ), + # ], + # reference_type=self.params.models.petrophysical_model.entity_type, + # ) + # ) directives_list.append( directives.SaveLPModelGroup( self.inversion_mesh.entity,