summaryrefslogtreecommitdiff
path: root/numpy/lib/arrayterator.py
diff options
context:
space:
mode:
Diffstat (limited to 'numpy/lib/arrayterator.py')
-rw-r--r--numpy/lib/arrayterator.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/numpy/lib/arrayterator.py b/numpy/lib/arrayterator.py
index d4a91b001..3f07cd263 100644
--- a/numpy/lib/arrayterator.py
+++ b/numpy/lib/arrayterator.py
@@ -14,6 +14,10 @@ from operator import mul
__all__ = ['Arrayterator']
+import sys
+if sys.version_info[0] >= 3:
+ from functools import reduce
+
class Arrayterator(object):
"""
Buffered iterator for big arrays.