summaryrefslogtreecommitdiff
path: root/numpy/fft/helper.py
diff options
context:
space:
mode:
Diffstat (limited to 'numpy/fft/helper.py')
-rw-r--r--numpy/fft/helper.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/numpy/fft/helper.py b/numpy/fft/helper.py
index 763d6684b..c22c92d76 100644
--- a/numpy/fft/helper.py
+++ b/numpy/fft/helper.py
@@ -60,7 +60,7 @@ def fftshift(x, axes=None):
tmp = asarray(x)
ndim = len(tmp.shape)
if axes is None:
- axes = range(ndim)
+ axes = list(range(ndim))
elif isinstance(axes, (int, nt.integer)):
axes = (axes,)
y = tmp
@@ -108,7 +108,7 @@ def ifftshift(x, axes=None):
tmp = asarray(x)
ndim = len(tmp.shape)
if axes is None:
- axes = range(ndim)
+ axes = list(range(ndim))
elif isinstance(axes, (int, nt.integer)):
axes = (axes,)
y = tmp