From 658b9c58f772006c83e6e1312aef05fee3b6c0ab Mon Sep 17 00:00:00 2001 From: Josh Carmichael Date: Thu, 13 Jun 2024 15:11:04 -0400 Subject: [PATCH] last cupy fill --- src/aspire/basis/fle_2d.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/aspire/basis/fle_2d.py b/src/aspire/basis/fle_2d.py index 4278331e1a..df1d66c608 100644 --- a/src/aspire/basis/fle_2d.py +++ b/src/aspire/basis/fle_2d.py @@ -452,7 +452,7 @@ def _create_basis_functions(self): """ Generate the actual basis functions as Python lambda operators """ - norm_constants = xp.zeros(self.count) + norm_constants = np.zeros(self.count) basis_functions = [None] * self.count for i in range(self.count): # parameters defining the basis function: bessel order and which bessel root @@ -481,7 +481,7 @@ def _create_basis_functions(self): norm_constants[i] = c - self.norm_constants = norm_constants + self.norm_constants = xp.asarray(norm_constants) self.basis_functions = basis_functions def _evaluate(self, coefs):