Skip to content

Commit 5e4e4a9

Browse files
committed
Support derived Animate classes
1 parent 858601d commit 5e4e4a9

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

spatialmath/base/graphics.py

+5-4
Original file line numberDiff line numberDiff line change
@@ -1462,10 +1462,11 @@ def _axes_dimensions(ax: plt.Axes) -> int:
14621462
else:
14631463
# handle the case of Animate objects pretending to be Axes
14641464
classname = ax.__class__.__name__
1465-
if classname == "Animate":
1466-
ret = 3
1467-
elif classname == "Animate2":
1468-
ret = 2
1465+
base_classes = ax.__class__.__bases__
1466+
if classname in ("Axes3DSubplot", "Animate") or any(base_class.__name__ in ("Axes3DSubplot", "Animate") for base_class in base_classes):
1467+
return 3
1468+
elif classname in ("AxesSubplot", "Animate2"):
1469+
return 2
14691470
# print("_axes_dimensions ", ax, ret)
14701471
return ret
14711472

0 commit comments

Comments
 (0)