make robust to type of passed arg

petercorke · petercorke · commit c77683697bba · 2023-03-06T17:20:03.000+10:00
diff --git a/spatialmath/base/graphics.py b/spatialmath/base/graphics.py
@@ -936,9 +936,10 @@ def ellipsoid(
             E = np.linalg.inv(E)
 
         x, y, z = sphere()  # unit sphere
+        centre = smb.getvector(centre, 3, out="col")
         e = (
             scale * sqrtm(E) @ np.array([x.flatten(), y.flatten(), z.flatten()])
-            + np.c_[centre]
+            + centre
         )
         return (
             e[0, :].reshape(x.shape),
