further changes to move to astropy

git-svn-id: http://svn.stsci.edu/svn/ssb/stsci_python/stwcs/trunk@31270 fe389314-cf27-0410-b35b-8c050e845b92

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