Commit b19d5d58 authored by Jait Dixit's avatar Jait Dixit
Browse files

Fix check when axes is None

parent 588237ae
Pipeline #4799 skipped
......@@ -378,8 +378,9 @@ class FFTW(FFT):
temp_val = None
# If axes tuple includes all axes, set it to None
if set(axes) == set(range(len(val.shape))):
axes = None
if axes is not None:
if set(axes) == set(range(len(val.shape))):
axes = None
for slice_list in\
utilities.get_slice_list(local_val.shape,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment