Fix test failures when radio_beam is installed, and also failure in _…

…_dir__ with Python 2
radio-astro-tools · Jan 7, 2016 · 5d77a74 · 5d77a74
1 parent 957e6b3
commit 5d77a74
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/spectral_cube/io/fits.py b/spectral_cube/io/fits.py
@@ -145,7 +145,7 @@ def load_fits_cube(input, hdu=0, meta=None, **kwargs):
         for component in data:
             comp_data, comp_wcs = cube_utils._orient(data[component], wcs)
             comp_mask = LazyMask(np.isfinite, data=comp_data, wcs=comp_wcs)
-            stokes_data[component] = SpectralCube(comp_data, wcs=comp_wcs, mask=comp_mask, meta=meta)
+            stokes_data[component] = SpectralCube(comp_data, wcs=comp_wcs, mask=comp_mask, meta=meta, header=header)
 
         cube = StokesSpectralCube(stokes_data)
 

diff --git a/spectral_cube/spectral_cube.py b/spectral_cube/spectral_cube.py
@@ -2332,7 +2332,7 @@ def wcs(self):
         return self._wcs
 
     def __dir__(self):
-        return list(self._stokes_data.keys()) + object.__dir__(self)
+        return list(self._stokes_data.keys()) + super(StokesSpectralCube, self).__dir__()
 
     def __getattr__(self, attribute):
         """