From 00ded4ab444e3d67216c9d54d8fbc10d1cf3cd9f Mon Sep 17 00:00:00 2001
From: Andrea Zonca <code@andreazonca.com>
Date: Mon, 27 Jul 2020 18:40:47 -0700
Subject: [PATCH] feat: import all datamodels into root package

---
 iris_pipeline/__init__.py            | 2 +-
 iris_pipeline/datamodels/__init__.py | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/iris_pipeline/__init__.py b/iris_pipeline/__init__.py
index 53e5bbc..4016e87 100644
--- 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
index c8a61c6..8e0977c 100644
--- 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}