Support sub-classes of spherical representations

astrofrog · Aug 20, 2015 · a826b6d · a826b6d
1 parent 8a208b8
commit a826b6d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/wcsaxes/transforms.py b/wcsaxes/transforms.py
@@ -244,8 +244,7 @@ def transform(self, input_coords):
 
         c_out = c_in.transform_to(self.output_system)
 
-        if (c_out.representation is SphericalRepresentation or
-            c_out.representation is UnitSphericalRepresentation):
+        if issubclass(c_out.representation, (SphericalRepresentation, UnitSphericalRepresentation)):
             lon = c_out.data.lon.deg
             lat = c_out.data.lat.deg
         else: