import eveything from jwst

iris_pipeline/__init__.py

@@ -37,5 +37,6 @@ class UnsupportedPythonError(Exception):
 from .parse_subarray_map import *
 from .merge_subarrays import *
 from .assign_wcs import *
+from .resample import *
 
 from .datamodels import  monkeypatch_jwst_datamodels

iris_pipeline/resample/resample.py

@@ -2,11 +2,12 @@
 from collections import OrderedDict
 import numpy as np
 
-from .. import datamodels
+# try first importing everything from JWST
+from jwst import datamodels
 
 from . import gwcs_drizzle
 from . import resample_utils
-from ..model_blender import blendmeta
+from jwst.model_blender import blendmeta
 
 log = logging.getLogger(__name__)
 log.setLevel(logging.DEBUG)

iris_pipeline/resample/resample_spec.py

@@ -11,7 +11,8 @@
 from gwcs import wcstools, WCS
 from gwcs import coordinate_frames as cf
 
-from .. import datamodels
+# try first importing everything from JWST
+from jwst import datamodels
 from . import gwcs_drizzle
 from . import resample_utils
 

iris_pipeline/resample/resample_spec_step.py

@@ -1,10 +1,12 @@
 __all__ = ["ResampleSpecStep"]
 
-from .. import datamodels
-from ..datamodels import MultiSlitModel, ModelContainer
+# try first importing everything from JWST
+from jwst import datamodels
+from jwst.datamodels import MultiSlitModel, ModelContainer
+
 from . import resample_spec, ResampleStep
-from ..exp_to_source import multislit_to_container
-from ..assign_wcs.util import update_s_region_spectral
+from jwst.exp_to_source import multislit_to_container
+from jwst.assign_wcs.util import update_s_region_spectral
 
 
 class ResampleSpecStep(ResampleStep):

iris_pipeline/resample/resample_step.py

@@ -2,12 +2,12 @@
 
 import numpy as np
 
-from ..stpipe import Step
-from ..extern.configobj.validate import Validator
-from ..extern.configobj.configobj import ConfigObj
-from .. import datamodels
+from jwst.stpipe import Step
+from jwst.extern.configobj.validate import Validator
+from jwst.extern.configobj.configobj import ConfigObj
+from jwst import datamodels
 from . import resample
-from ..assign_wcs import util
+from jwst.assign_wcs import util
 
 log = logging.getLogger(__name__)
 log.setLevel(logging.DEBUG)

iris_pipeline/resample/resample_utils.py

@@ -8,7 +8,7 @@
 
 from astropy.nddata.bitmask import interpret_bit_flags
 
-from ..assign_wcs.util import wcs_from_footprints, wcs_bbox_from_shape
+from jwst.assign_wcs.util import wcs_from_footprints, wcs_bbox_from_shape
 
 import logging
 log = logging.getLogger(__name__)