summaryrefslogtreecommitdiff
path: root/numpy/lib
diff options
context:
space:
mode:
Diffstat (limited to 'numpy/lib')
-rw-r--r--numpy/lib/arrayterator.py3
-rw-r--r--numpy/lib/tests/test_arrayterator.py4
2 files changed, 2 insertions, 5 deletions
diff --git a/numpy/lib/arrayterator.py b/numpy/lib/arrayterator.py
index b71e9a123..360d61bc5 100644
--- a/numpy/lib/arrayterator.py
+++ b/numpy/lib/arrayterator.py
@@ -14,8 +14,7 @@ from operator import mul
__all__ = ['Arrayterator']
import sys
-if sys.version_info[0] >= 3:
- from functools import reduce
+from functools import reduce
class Arrayterator(object):
"""
diff --git a/numpy/lib/tests/test_arrayterator.py b/numpy/lib/tests/test_arrayterator.py
index 28feb4824..3fa15fef1 100644
--- a/numpy/lib/tests/test_arrayterator.py
+++ b/numpy/lib/tests/test_arrayterator.py
@@ -1,15 +1,13 @@
from __future__ import division, absolute_import
from operator import mul
+from functools import reduce
import numpy as np
from numpy.random import randint
from numpy.lib import Arrayterator
from numpy.testing import assert_
-import sys
-if sys.version_info[0] >= 3:
- from functools import reduce
def test():
np.random.seed(np.arange(10))