add stuff to __all__ when doing import_and_attach_all()

LSSTDESC · Jul 15, 2024 · 26c2ce8 · 26c2ce8
1 parent 0222836
commit 26c2ce8
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/src/rail/stages/__init__.py b/src/rail/stages/__init__.py
@@ -34,13 +34,6 @@
 
 from rail.tools.table_tools import ColumnMapper, RowSelector, TableConverter
 
-
-def import_and_attach_all():
-    """Import all the packages in the rail ecosystem and attach them to this module"""
-    RailEnv.import_all_packages()
-    RailEnv.attach_stages(rail.stages)
-
-
 __all__ = [
     "CatEstimator",
     "CatClassifier",
@@ -79,3 +72,12 @@ def import_and_attach_all():
     "RowSelector",
     "TableConverter",
 ]
+
+
+def import_and_attach_all():
+    """Import all the packages in the rail ecosystem and attach them to this module"""
+    RailEnv.import_all_packages()
+    RailEnv.attach_stages(rail.stages)
+    for xx in dir(rail.stages):
+        rail.stages.__all__.append(xx)
+