feat: import all datamodels into root package

oirlab · Sep 16, 2020 · 00ded4a · 00ded4a
1 parent a30d7f1
commit 00ded4a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/iris_pipeline/__init__.py b/iris_pipeline/__init__.py
@@ -39,4 +39,4 @@ class UnsupportedPythonError(Exception):
 from .assign_wcs import *
 from .resample import *
 
-from .datamodels import  monkeypatch_jwst_datamodels
+from .datamodels import  *
diff --git a/iris_pipeline/datamodels/__init__.py b/iris_pipeline/datamodels/__init__.py
@@ -14,6 +14,7 @@
 
 
 __all__ = [
+    "monkeypatch_jwst_datamodels",
     "IRISImageModel",
     "TMTRampModel",
     "TMTMaskModel",
@@ -25,7 +26,7 @@
     "TMTReferenceFileModel",
 ]
 
-_all_models = __all__
+_all_models = __all__[1:]
 _local_dict = locals()
 _defined_models = {k: _local_dict[k] for k in _all_models}