fix import in test/docs

(needed for "move code to allow importing slice mixins without circular imports")
spacetelescope · Jul 8, 2024 · 8a3fc8f · 8a3fc8f
1 parent 611eb50
commit 8a3fc8f
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/docs/reference/api_nuts_bolts.rst b/docs/reference/api_nuts_bolts.rst
@@ -23,6 +23,10 @@ Utilities API
    :no-inheritance-diagram:
    :no-inherited-members:
 
+.. automodapi:: jdaviz.configs.cubeviz.plugins.mixins
+   :no-inheritance-diagram:
+   :no-inherited-members:
+
 .. automodapi:: jdaviz.configs.imviz.wcs_utils
    :no-inheritance-diagram:
    :no-inherited-members:

diff --git a/jdaviz/configs/imviz/plugins/tools.py b/jdaviz/configs/imviz/plugins/tools.py
@@ -135,7 +135,7 @@ def _dragevent(self, data):
             self._layer_state.contrast = contrast
 
     def on_mouse_or_key_event(self, data):
-        from jdaviz.configs.imviz.helper import get_top_layer_index
+        from jdaviz.utils import get_top_layer_index
 
         event = data['event']
 

diff --git a/jdaviz/configs/imviz/tests/test_viewer_tools.py b/jdaviz/configs/imviz/tests/test_viewer_tools.py
@@ -1,6 +1,6 @@
 from numpy.testing import assert_allclose
 
-from jdaviz.configs.imviz.helper import get_top_layer_index
+from jdaviz.utils import get_top_layer_index
 from jdaviz.configs.imviz.tests.utils import BaseImviz_WCS_NoWCS