Fix same test bug

FalkonML · Apr 18, 2024 · b0853d6 · b0853d6
1 parent 0c5388c
commit b0853d6
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