Skip to content

Commit 09b0c43

Browse files
authored
fix: support derived Animate classes in order to use the same axes. (#50)
1 parent 6e1edea commit 09b0c43

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)