Use 'with' suggestion.

lsst · Aug 24, 2021 · 5e03d26 · 5e03d26
1 parent d21b2cb
commit 5e03d26
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/python/lsst/obs/subaru/strayLight/yStrayLight.py b/python/lsst/obs/subaru/strayLight/yStrayLight.py
@@ -158,15 +158,11 @@ class SubaruStrayLightData(StrayLightData):
     @classmethod
     def readFits(cls, filename, **kwargs):
         calib = cls()
-        hdulist = fits.open(filename)
 
-        calib.ampData = []
-        for extension in (0, 1, 2, 3):
-            calib.ampData.append(hdulist[extension].data)
+        with fits.open(filename) as hdulist:
+            calib.ampData = [hdu.data for hdu in hdulist]
+            calib.setMetadata(hdulist[0].header)
 
-        calib.setMetadata(hdulist[0].header)
-
-        hdulist.close()
         calib.log.info("Finished reading straylightData.")
         return calib