fix import orders

src/arviz/__init__.py

@@ -8,8 +8,8 @@
 info = ""
 
 try:
-    import arviz_base as base
     from arviz_base import *
+    import arviz_base as base
 
     _status = (
         f"arviz_base {base.__version__} available, "
@@ -26,9 +26,13 @@
 info += _status + "\n"
 
 try:
-    import arviz_stats as stats
     from arviz_stats import *
 
+    # the base computational module fron arviz_stats will override the alias to arviz-base
+    # arviz.stats.base will still be available
+    import arviz_base as base
+    import arviz_stats as stats
+
     _status = (
         f"arviz_stats {getattr(stats, '__version__', '0.0')} available, "
         "exposing its functions as part of the `arviz` namespace"
@@ -43,8 +47,8 @@
 info += _status + "\n"
 
 try:
-    import arviz_plots as plots
     from arviz_plots import *
+    import arviz_plots as plots
 
     _status = (
         f"arviz_plots {plots.__version__} available, "

tests/test_namespace.py

@@ -23,3 +23,21 @@ def test_aliases():
                 assert False, obj_name
             assert orig_lib.startswith("arviz"), obj_name
             assert orig_lib != "arviz", obj_name
+
+
+def test_base_alias():
+    import arviz_base
+
+    assert az.base is arviz_base
+
+
+def test_stats_alias():
+    import arviz_stats
+
+    assert az.stats is arviz_stats
+
+
+def test_plots_alias():
+    import arviz_plots
+
+    assert az.plots is arviz_plots