_like to avoid quantities

spacetelescope · May 22, 2024 · 9231d1d · 9231d1d
1 parent 799b771
commit 9231d1d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/jdaviz/configs/cubeviz/plugins/spectral_extraction/spectral_extraction.py b/jdaviz/configs/cubeviz/plugins/spectral_extraction/spectral_extraction.py
@@ -315,7 +315,7 @@ def aperture_weight_mask(self):
         # wavelength, on the range [0, 1].
         if self.aperture.selected == self.aperture.default_text:
             # Entire Cube
-            return np.ones_like(self.dataset.selected_obj.flux)
+            return np.ones_like(self.dataset.selected_obj.flux.value)
         return self.aperture.get_mask(self.dataset.selected_obj,
                                       self.aperture_method_selected,
                                       self.spectral_display_unit,
@@ -325,7 +325,7 @@ def aperture_weight_mask(self):
     def bg_weight_mask(self):
         if self.background.selected == self.background.default_text:
             # NO background
-            return np.zeros_like(self.dataset.selected_obj.flux)
+            return np.zeros_like(self.dataset.selected_obj.flux.value)
         return self.background.get_mask(self.dataset.selected_obj,
                                         self.aperture_method_selected,
                                         self.spectral_display_unit,