Fix a naxes bug

spacetelescope · Nov 20, 2024 · 19d024b · 19d024b
1 parent dd5e514
commit 19d024b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gwcs/api.py b/gwcs/api.py
@@ -65,7 +65,7 @@ def world_axis_units(self):
 
     def _remove_quantity_output(self, result, frame):
         if self.forward_transform.uses_quantity:
-            if self.output_frame.naxes == 1:
+            if frame.naxes == 1:
                 result = [result]
 
             result = tuple(r.to_value(unit) if isinstance(r, u.Quantity) else r