From c857529ce1145744b9f917d43a510906f2551a72 Mon Sep 17 00:00:00 2001 From: Josh Carmichael Date: Tue, 11 Jun 2024 15:24:56 -0400 Subject: [PATCH] convert pf to complex --- src/aspire/basis/ffb_2d.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/aspire/basis/ffb_2d.py b/src/aspire/basis/ffb_2d.py index 4971b44f77..821800ee54 100644 --- a/src/aspire/basis/ffb_2d.py +++ b/src/aspire/basis/ffb_2d.py @@ -167,7 +167,9 @@ def _evaluate(self, v): # perform inverse non-uniformly FFT transform back to 2D coordinate basis freqs = m_reshape(self._precomp["freqs"], (2, n_r * n_theta)) - x = 2 * anufft(pf, 2 * pi * freqs, self.sz, real=True) + x = 2 * anufft( + pf.astype(complex_type(self.dtype)), 2 * pi * freqs, self.sz, real=True + ) # Return X as Image instance with the last two dimensions as *self.sz x = x.reshape((*sz_roll, *self.sz))