Workaround for islice int error in animation.py (#8984)

braaannigan · tacaswell · commit 1da7f4b0e990 · 2017-08-08T11:21:59.000-04:00
* workaround for islice int error

* changed position of int conversion

* added self

* move into if statement

* added save_count assignment back in

* remove period

* shorten line to &lt; 80
diff --git a/lib/matplotlib/animation.py b/lib/matplotlib/animation.py
@@ -1544,7 +1544,6 @@ def __init__(self, fig, func, frames=None, init_func=None, fargs=None,
         # used if we don't know how many frames there will be: in the case
         # of no generator or in the case of a callable.
         self.save_count = save_count
-
         # Set up a function that creates a new iterable when needed. If nothing
         # is passed in for frames, just use itertools.count, which will just
         # keep counting from 0. A callable passed in for frames is assumed to
@@ -1562,9 +1561,15 @@ def __init__(self, fig, func, frames=None, init_func=None, fargs=None,
             self._iter_gen = lambda: iter(xrange(frames))
             self.save_count = frames
 
-        # If we're passed in and using the default, set it to 100.
+        # If we're passed in and using the default, set save_count to 100.
+        # itertools.islice can return an error when passed a numpy int instead
+        # of a native python int. This is a known issue:
+        # http://bugs.python.org/issue30537
+        # As a workaround, convert save_count to native python int
         if self.save_count is None:
             self.save_count = 100
+        else:
+            self.save_count = int(self.save_count)
 
         self._init_func = init_func
 
