Skip to content

Commit fbbcc9e

Browse files
tahsinkosepetercorke
authored andcommitted
Support derived Animate classes
1 parent fff7009 commit fbbcc9e

1 file changed

Lines changed: 5 additions & 4 deletions

File tree

spatialmath/base/graphics.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1512,10 +1512,11 @@ def _axes_dimensions(ax: plt.Axes) -> int:
15121512
else:
15131513
# handle the case of Animate objects pretending to be Axes
15141514
classname = ax.__class__.__name__
1515-
if classname == "Animate":
1516-
ret = 3
1517-
elif classname == "Animate2":
1518-
ret = 2
1515+
base_classes = ax.__class__.__bases__
1516+
if classname in ("Axes3DSubplot", "Animate") or any(base_class.__name__ in ("Axes3DSubplot", "Animate") for base_class in base_classes):
1517+
return 3
1518+
elif classname in ("AxesSubplot", "Animate2"):
1519+
return 2
15191520
# print("_axes_dimensions ", ax, ret)
15201521
return ret
15211522

0 commit comments

Comments
 (0)