Fix output shape for Orbit.theta

jobovy · May 24, 2024 · 1745307 · 1745307
1 parent c976e79
commit 1745307
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/galpy/orbit/Orbits.py b/galpy/orbit/Orbits.py
@@ -4092,7 +4092,7 @@ def theta(self, *args, **kwargs):
         if self.dim() != 3:
             raise AttributeError("Orbit must be 3D to use theta()")
         else:
-            return numpy.arctan2(thiso[0], thiso[3])
+            return numpy.arctan2(thiso[0], thiso[3]).T
 
     @physical_conversion("angle_deg")
     @shapeDecorator