import GriddedPSFModel from photutils.psf

spacetelescope · Oct 23, 2023 · ab1d205 · ab1d205
1 parent 1430b92
commit ab1d205
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/webbpsf/gridded_library.py b/webbpsf/gridded_library.py
@@ -427,9 +427,12 @@ def to_model(data, meta):
             and oversampling keys
         """
         try:
-            from photutils import GriddedPSFModel
+            from photutils.psf import GriddedPSFModel
         except ImportError:
-            raise ImportError("This method requires photutils >= 0.6")
+            try:
+                from photutils import GriddedPSFModel
+            except ImportError:
+                raise ImportError("This method requires photutils >= 0.6")
 
         ndd = NDData(data, meta=meta, copy=True)
 

diff --git a/webbpsf/utils.py b/webbpsf/utils.py
@@ -834,9 +834,12 @@ def to_griddedpsfmodel(HDUlist_or_filename=None, ext_data=0, ext_header=0):
         grid_xypos and oversampling keys
     """
     try:
-        from photutils import GriddedPSFModel
+        from photutils.psf import GriddedPSFModel
     except ImportError:
-        raise ImportError("This method requires photutils >= 0.6")
+        try:
+            from photutils import GriddedPSFModel
+        except ImportError:
+            raise ImportError("This method requires photutils >= 0.6")
 
     if isinstance(HDUlist_or_filename, str):
         HDUlist = fits.open(HDUlist_or_filename)