code cleanup

spacetelescope · Apr 17, 2023 · 83a1710 · 83a1710
1 parent 7c8b225
commit 83a1710
Showing 1 changed file with 7 additions and 9 deletions.
diff --git a/jdaviz/core/marks.py b/jdaviz/core/marks.py
@@ -68,7 +68,13 @@ def __init__(self, *args, **kwargs):
         self.auto_update_units = True
         self.hub.subscribe(self, GlobalDisplayUnitChanged,
                            handler=self._on_global_display_unit_changed)
-        self._update_units()
+
+        if not self.auto_update_units:
+            return
+        if self.xunit is None:
+            self.set_x_unit()
+        if self.yunit is None:
+            self.set_y_unit()
 
     @property
     def hub(self):
@@ -82,14 +88,6 @@ def append_xy(self, x, y):
         self.x = np.append(self.x, x)
         self.y = np.append(self.y, y)
 
-    def _update_units(self):
-        if not self.auto_update_units:
-            return
-        if self.xunit is None:
-            self.set_x_unit()
-        if self.yunit is None:
-            self.set_y_unit()
-
     def set_x_unit(self, unit=None):
         if unit is None:
             if not hasattr(self.viewer.state, 'x_display_unit'):