Commit 5adfb1c8 authored by theos's avatar theos
Browse files

Fixed a bug with spaces vs. axes. in FFTOperator.

parent cb713171
......@@ -40,7 +40,8 @@ class FFTOperator(LinearOperator):
def _times(self, x, spaces, types):
spaces = utilities.cast_axis_to_tuple(spaces, len(x.domain))
new_val = self._forward_transformation.transform(x.val, axes=spaces)
axes = x.domain_axes[spaces[0]]
new_val = self._forward_transformation.transform(x.val, axes=axes)
if spaces is None:
result_domain = self.target
......@@ -56,7 +57,8 @@ class FFTOperator(LinearOperator):
def _inverse_times(self, x, spaces, types):
spaces = utilities.cast_axis_to_tuple(spaces, len(x.domain))
new_val = self._inverse_transformation.transform(x.val, axes=spaces)
axes = x.domain_axes[spaces[0]]
new_val = self._inverse_transformation.transform(x.val, axes=axes)
if spaces is None:
result_domain = self.domain
......
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