Use astropy rather than pyfits

lsst-dm · Jul 10, 2018 · 10d640f · 10d640f
1 parent 8e4db6e
commit 10d640f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 9 deletions.
diff --git a/python/lsst/meas/mosaic/checkMosaicTask.py b/python/lsst/meas/mosaic/checkMosaicTask.py
@@ -384,17 +384,17 @@ def plotPosAsMag(self, m0_s, dx_s, dy_s):
     def writeCatalog(self, allSource, wcsDic, calibDic, ffpDic):
         num = sum(sum(1 for src in ss if src.getExp() >=0 and src.getChip() >= 0) for ss in allSource)
 
-        import pyfits
-
-        schema = pyfits.ColDefs([pyfits.Column(name="id", format="K"),
-                                 pyfits.Column(name="ra", format="D"),
-                                 pyfits.Column(name="dec", format="D"),
-                                 pyfits.Column(name="mag", format="E"),
-                                 pyfits.Column(name="err", format="E"),
-                                 pyfits.Column(name="corr", format="E"),
+        from astropy.io import fits
+
+        schema = fits.ColDefs([fits.Column(name="id", format="K"),
+                                 fits.Column(name="ra", format="D"),
+                                 fits.Column(name="dec", format="D"),
+                                 fits.Column(name="mag", format="E"),
+                                 fits.Column(name="err", format="E"),
+                                 fits.Column(name="corr", format="E"),
                                  ])
 
-        outHdu = pyfits.new_table(schema, nrows=num)
+        outHdu = fits.new_table(schema, nrows=num)
         outData = outHdu.data
 
         i = 0

diff --git a/ups/meas_mosaic.table b/ups/meas_mosaic.table
@@ -14,6 +14,7 @@ setupRequired(meas_astrom)
 setupRequired(meas_base)
 setupRequired(meas_algorithms)
 setupRequired(minuit2)
+setupRequired(astropy)
 
 setupOptional(obs_subaru)