re-initialize to account for spatial subset in cubeviz

see #1954
spacetelescope · Jan 3, 2023 · efdae47 · efdae47
1 parent 9f9350e
commit efdae47
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jdaviz/configs/default/plugins/model_fitting/model_fitting.py b/jdaviz/configs/default/plugins/model_fitting/model_fitting.py
@@ -477,7 +477,7 @@ def _initialize_model_component(self, model_comp, comp_label, poly_order=None):
 
             initial_values[param_name] = initial_val
 
-        masked_spectrum = self._apply_subset_masks(self.dataset.selected_obj, self.spectral_subset)
+        masked_spectrum = self._apply_subset_masks(self._get_1d_spectrum(), self.spectral_subset)
         mask = masked_spectrum.mask
         initialized_model = initialize(
             MODELS[model_comp](name=comp_label,