diff options
Diffstat (limited to 'lib/stwcs/updatewcs/det2im.py')
| -rw-r--r-- | lib/stwcs/updatewcs/det2im.py | 18 |
1 files changed, 9 insertions, 9 deletions
diff --git a/lib/stwcs/updatewcs/det2im.py b/lib/stwcs/updatewcs/det2im.py index a783c88..902deaa 100644 --- a/lib/stwcs/updatewcs/det2im.py +++ b/lib/stwcs/updatewcs/det2im.py @@ -1,6 +1,6 @@ from __future__ import division # confidence high -from astropy.io import fits as pyfits +from astropy.io import fits from stsci.tools import fileutil import utils import numpy as np @@ -33,15 +33,15 @@ class DET2IMCorr(object): """ Parameters ---------- - fobj: pyfits object + fobj: `astropy.io.fits.HDUList` object Science file, for which a distortion correction in a NPOLFILE is available """ logger.info("\n\tStarting DET2IM: %s" %time.asctime()) try: - assert isinstance(fobj, pyfits.HDUList) + assert isinstance(fobj, fits.HDUList) except AssertionError: - logger.exception('\n\tInput must be a pyfits.HDUList object') + logger.exception('\n\tInput must be a fits.HDUList object') raise cls.applyDet2ImCorr(fobj) @@ -54,7 +54,7 @@ class DET2IMCorr(object): def applyDet2ImCorr(cls, fobj): """ - For each science extension in a pyfits file object: + For each science extension in a fits file object: - create a WCSDVARR extension - update science header - add/update D2IMEXT keyword @@ -159,7 +159,7 @@ class DET2IMCorr(object): Make sure 'CCDCHIP' in the npolfile matches "CCDCHIP' in the science file. """ xdata, ydata = (None, None) - d2im = pyfits.open(d2imfile) + d2im = fits.open(d2imfile) for ext in d2im: d2imextname = ext.header.get('EXTNAME',"") d2imccdchip = ext.header.get('CCDCHIP',1) @@ -183,7 +183,7 @@ class DET2IMCorr(object): hdr = cls.createD2ImHdr(sciheader, d2imfile=d2imfile, wdvarr_ver=wdvarr_ver, d2im_extname=d2im_extname, ccdchip=ccdchip) - hdu=pyfits.ImageHDU(header=hdr, data=data) + hdu = fits.ImageHDU(header=hdr, data=data) return hdu createD2ImHDU = classmethod(createD2ImHDU) @@ -196,7 +196,7 @@ class DET2IMCorr(object): is such that a full size d2im table is created and then shifted or scaled if the science image is a subarray or binned image. """ - d2im = pyfits.open(d2imfile) + d2im = fits.open(d2imfile) d2im_phdr = d2im[0].header for ext in d2im: try: @@ -272,7 +272,7 @@ class DET2IMCorr(object): for card in d2im_phdr.cards[start_indx:end_indx]: cdl.append(card) - hdr = pyfits.Header(cards=cdl) + hdr = fits.Header(cards=cdl) return hdr |