Merge pull request #14048 from timhoffm/fix-animation-save

tacaswell · web-flow · commit 63ef1a1ad414 · 2019-04-26T08:45:54.000-04:00
FIX: Animation.save
diff --git a/lib/matplotlib/animation.py b/lib/matplotlib/animation.py
@@ -1134,7 +1134,10 @@ def func(current_frame: int, total_frames: int) -> Any
                     # Clear the initial frame
                     anim._init_draw()
                 frame_number = 0
-                save_count_list = [a.save_count for a in all_anim]
+                # TODO: Currently only FuncAnimation has a save_count
+                #       attribute. Can we generalize this to all Animations?
+                save_count_list = [getattr(a, 'save_count', None)
+                                   for a in all_anim]
                 if None in save_count_list:
                     total_frames = None
                 else:
