fix setting display units for unitless data

spacetelescope · Apr 13, 2023 · a41c92f · a41c92f
1 parent c4bb23d
commit a41c92f
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/jdaviz/configs/specviz/plugins/viewers.py b/jdaviz/configs/specviz/plugins/viewers.py
@@ -361,8 +361,10 @@ def add_data(self, data, color=None, alpha=None, **layer_state):
         result = super().add_data(data, color, alpha, **layer_state)
 
         if reset_plot_axes:
-            self.state.x_display_unit = data.get_component(self.state.x_att.label).units
-            self.state.y_display_unit = data.get_component("flux").units
+            x_units = data.get_component(self.state.x_att.label).units
+            y_units = data.get_component("flux").units
+            self.state.x_display_unit = x_units if len(x_units) else None
+            self.state.y_display_unit = y_units if len(y_units) else None
             self.set_plot_axes()
 
         self._plot_uncertainties()
@@ -489,7 +491,8 @@ def _plot_uncertainties(self):
     def set_plot_axes(self):
         # Set axes labels for the spectrum viewer
         flux_unit_type = "Flux density"
-        x_unit = u.Unit(self.state.x_display_unit)
+        x_disp_unit = self.state.x_display_unit
+        x_unit = u.Unit(x_disp_unit) if x_disp_unit is not None else u.dimensionless_unscaled
         if x_unit.is_equivalent(u.m):
             spectral_axis_unit_type = "Wavelength"
         elif x_unit.is_equivalent(u.Hz):