replace deprecated GriddedPSFModel import

spacetelescope · Jul 26, 2023 · af28236 · af28236
1 parent b219a7d
commit af28236
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/webbpsf/gridded_library.py b/webbpsf/gridded_library.py
@@ -112,9 +112,12 @@ def __init__(self, instrument, filter_name, detectors="all", num_psfs=16, psf_lo
 
         # Before doing anything else, check that we have GriddedPSFModel
         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")
 
         # Pull WebbPSF instance
         self.webb = instrument