Skip to content

Commit 5a61ff3

Browse files
committed
vol[0] in place of np.squeeze
1 parent c71b044 commit 5a61ff3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/aspire/reconstruction/estimator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,7 @@ def apply_kernel(self, vol_coeff, kernel=None):
148148
vol = self.basis.evaluate(vol_coeff)
149149
# convolve_volume expects a 3-dimensional array
150150
# so we remove the first dimension of the volume, which is 1
151-
vol = Volume(kernel.convolve_volume(np.squeeze(vol.asnumpy(), axis=0)))
151+
vol = kernel.convolve_volume(vol[0])
152152
vol = self.basis.evaluate_t(vol)
153153

154154
return vol

0 commit comments

Comments
 (0)