Skip to content

Commit 1d66a57

Browse files
committed
Remove ghost cells from model save. Add back app res
1 parent 719eb27 commit 1d66a57

3 files changed

Lines changed: 13 additions & 1 deletion

File tree

simpeg_drivers/components/data.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -344,6 +344,16 @@ def create_survey(self):
344344
)
345345
survey.cells = self.entity.cells
346346

347+
observed = self.entity.get_data("Observed_potential")
348+
if observed:
349+
self.entity.add_data(
350+
{
351+
"Observed_apparent_resistivity": {
352+
"values": survey.apparent_resistivity * observed[0].values
353+
}
354+
}
355+
)
356+
347357
if "induced polarization" in self.params.inversion_type:
348358
survey.cells = self.entity.cells
349359

simpeg_drivers/components/factories/directives_factory.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -436,7 +436,7 @@ def assemble_keyword_arguments(
436436
if self.params.models.model_type == ModelTypeEnum.resistivity:
437437
kwargs["transforms"].append(lambda x: 1 / x)
438438

439-
if "1d" in self.factory_type:
439+
if "1d" in self.factory_type or "2d" in self.factory_type:
440440
ghosts = (
441441
np.squeeze(np.asarray(inversion_object.permutation.sum(axis=0))) == 0
442442
)

simpeg_drivers/components/models.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -643,6 +643,8 @@ def _get_value(self, model: float | NumericData) -> np.ndarray:
643643
elif isinstance(model, int | float):
644644
nc = self.driver.inversion_mesh.mesh.n_cells
645645
model *= np.ones(nc)
646+
elif isinstance(model, np.ndarray):
647+
model = (self.driver.inversion_mesh.permutation @ model).astype(model.dtype)
646648

647649
return model
648650

0 commit comments

Comments
 (0)