attempt fix

Author: cvanelteren

ultraplot/__init__.py

@@ -26,6 +26,7 @@
 
 # Exceptions to the automated lazy loading
 _LAZY_LOADING_EXCEPTIONS = {
+    "constructor": ("constructor", None),
     "crs": ("proj", None),
     "colormaps": ("colors", "_cmap_database"),
     "check_for_update": ("utils", "check_for_update"),
@@ -39,7 +40,9 @@
     "tests": ("tests", None),
     "rcsetup": ("internals", "rcsetup"),
     "warnings": ("internals", "warnings"),
-    "figure": ("figure", "Figure"),
+    "Figure": ("figure", "Figure"),
+    "Axes": ("axes", "Axes"),
+    "rc": ("config", "rc"),
 }
 
 
@@ -229,7 +232,7 @@ def _load_all():
         _build_registry_map()
     if _REGISTRY_ATTRS:
         names.update(_REGISTRY_ATTRS)
-    names.update({"__version__", "version", "name"})
+    names.update({"__version__", "version", "name", "setup", "pyplot"})
     _EAGER_DONE = True
     return sorted(names)
 
@@ -264,7 +267,7 @@ def setup(*, eager=None):
 
 
 def __getattr__(name):
-    if name in {"pytest_plugins", "__version__", "version", "name", "__all__"}:
+    if name in {"__version__", "version", "name", "__all__"}:
         if name == "__all__":
             value = _load_all()
             globals()["__all__"] = value
@@ -297,14 +300,12 @@ def __getattr__(name):
 
     if name in _LAZY_LOADING_EXCEPTIONS:
         _setup()
-        _maybe_eager_import()
         return _resolve_extra(name)
 
     _discover_modules()
     if _ATTR_MAP and name in _ATTR_MAP:
         module_name, attr_name = _ATTR_MAP[name]
         _setup()
-        _maybe_eager_import()
 
         module = _import_module(module_name)
         value = getattr(module, attr_name) if attr_name else module

ultraplot/tests/test_imshow.py

@@ -1,8 +1,9 @@
+import numpy as np
 import pytest
-
-import ultraplot as plt, numpy as np
 from matplotlib.testing import setup
 
+import ultraplot as plt
+
 
 @pytest.fixture()
 def setup_mpl():
@@ -39,7 +40,6 @@ def test_standardized_input(rng):
         axs[1].pcolormesh(xedges, yedges, data)
         axs[2].contourf(x, y, data)
         axs[3].contourf(xedges, yedges, data)
-    fig.show()
     return fig