Merge branch 'master' of github.com:FalkonML/falkon

FalkonML · Apr 19, 2024 · fe672ec · fe672ec
2 parents 645e8f1 + b0853d6
commit fe672ec
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/falkon/tests/test_conjgrad.py b/falkon/tests/test_conjgrad.py
@@ -154,8 +154,8 @@ def test_restarts(self, data, centers, kernel, preconditioner, knm, kmm, vec_rhs
     def test_precomputed_kernel(self, data, centers, kernel, preconditioner, knm, kmm, vec_rhs, device):
         preconditioner = preconditioner.to(device)
         options = dataclasses.replace(self.basic_opt, use_cpu=device == "cpu")
-        knm = move_tensor(knm, device)
-        calc_kernel = PrecomputedKernel(knm, options)
+        knm_dev = move_tensor(knm, device)
+        calc_kernel = PrecomputedKernel(knm_dev, options)
         opt = FalkonConjugateGradient(calc_kernel, preconditioner, opt=options)
 
         # Solve (knm.T @ knm + lambda*n*kmm) x = knm.T @ b