diff --git a/simpeg_drivers/components/factories/survey_factory.py b/simpeg_drivers/components/factories/survey_factory.py index cfe38a7b..1169633e 100644 --- a/simpeg_drivers/components/factories/survey_factory.py +++ b/simpeg_drivers/components/factories/survey_factory.py @@ -214,8 +214,8 @@ def _add_data(self, survey, data, local_index, channel): dobs.append(data_stack[component_id][channel_id, rx_id]) uncerts.append(uncert_stack[component_id][channel_id, rx_id]) - survey.dobs = np.vstack([dobs]).flatten() - survey.std = np.vstack([uncerts]).flatten() + data_vec = np.vstack([dobs]).flatten() + uncertainty_vec = np.vstack([uncerts]).flatten() elif self.factory_type in ["magnetotellurics", "tipper"]: local_data = {} @@ -235,12 +235,6 @@ def _add_data(self, survey, data, local_index, channel): data_vec = self._stack_channels(local_data, "row") uncertainty_vec = self._stack_channels(local_uncertainties, "row") - uncertainty_vec[np.isnan(data_vec)] = np.inf - data_vec[np.isnan(data_vec)] = ( - self.dummy - ) # Nan's handled by inf uncertainties - survey.dobs = data_vec - survey.std = uncertainty_vec else: local_data = {k: v[local_index] for k, v in data.observed.items()} @@ -250,12 +244,11 @@ def _add_data(self, survey, data, local_index, channel): data_vec = self._stack_channels(local_data, "column") uncertainty_vec = self._stack_channels(local_uncertainties, "column") - uncertainty_vec[np.isnan(data_vec)] = np.inf - data_vec[np.isnan(data_vec)] = ( - self.dummy - ) # Nan's handled by inf uncertainties - survey.dobs = data_vec - survey.std = uncertainty_vec + + uncertainty_vec[np.isnan(data_vec)] = np.inf + data_vec[np.isnan(data_vec)] = self.dummy # Nan's handled by inf uncertainties + survey.dobs = data_vec + survey.std = uncertainty_vec def _stack_channels(self, channel_data: dict[str, np.ndarray], mode: str): """ diff --git a/simpeg_drivers/electromagnetics/time_domain/params.py b/simpeg_drivers/electromagnetics/time_domain/params.py index e193526b..fb4623c4 100644 --- a/simpeg_drivers/electromagnetics/time_domain/params.py +++ b/simpeg_drivers/electromagnetics/time_domain/params.py @@ -54,7 +54,7 @@ def __init__(self, input_file=None, forward_only=False, **kwargs): self._y_channel_bool = None self._y_channel = None self._y_uncertainty = None - self._inversion_type = "tipper" + self._model_type = "Conductivity (S/m)" super().__init__(input_file=input_file, forward_only=forward_only, **kwargs)