Support sub-classes of spherical representations

astrofrog · Aug 20, 2015 · 0eab81c · 0eab81c
1 parent 9b0c2fc
commit 0eab81c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/wcsaxes/transforms.py b/wcsaxes/transforms.py
@@ -245,8 +245,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: