Commit 06e2d7fa authored by Jait Dixit's avatar Jait Dixit
Browse files

Fix get_centering_mask calls

parent 38aa1c4b
...@@ -451,11 +451,13 @@ class FFTWTransformInfo(object): ...@@ -451,11 +451,13 @@ class FFTWTransformInfo(object):
self.cmask_domain = fftw_context.get_centering_mask( self.cmask_domain = fftw_context.get_centering_mask(
domain.zerocenter, domain.zerocenter,
local_shape if axes is None else
[y for x, y in enumerate(local_shape) if x in axes], [y for x, y in enumerate(local_shape) if x in axes],
local_offset_Q) local_offset_Q)
self.cmask_codomain = fftw_context.get_centering_mask( self.cmask_codomain = fftw_context.get_centering_mask(
codomain.zerocenter, codomain.zerocenter,
local_shape if axes is None else
[y for x, y in enumerate(local_shape) if x in axes], [y for x, y in enumerate(local_shape) if x in axes],
local_offset_Q) local_offset_Q)
......
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