Merge pull request #45 from astrofrog/gwcs-spec

Add back use of SpectralCoordinate to wrap GWCS for now

glue_astronomy/translators/spectrum1d.py

@@ -3,6 +3,8 @@
 from glue.config import data_translator
 from glue.core import Data, Subset
 
+from gwcs import WCS as GWCS
+
 from astropy.wcs import WCS
 from astropy import units as u
 from astropy.wcs import WCSSUB_SPECTRAL
@@ -122,7 +124,9 @@ def to_data(self, obj):
             data['flux'] = np.swapaxes(obj.flux, -1, 0)
             data.get_component('flux').units = str(obj.unit)
         else:
-            if obj.flux.ndim == 2 and obj.wcs.world_n_dim == 1:
+            if obj.flux.ndim == 1 and obj.wcs.world_n_dim == 1 and isinstance(obj.wcs, GWCS):
+                data = Data(coords=SpectralCoordinates(obj.spectral_axis))
+            elif obj.flux.ndim == 2 and obj.wcs.world_n_dim == 1:
                 data = Data(coords=PaddedSpectrumWCS(obj.wcs))
             else:
                 data = Data(coords=obj.wcs)