test changed to 'cpu'

lbl-camera · Feb 23, 2023 · 70b28a1 · 70b28a1
1 parent bac2ec0
commit 70b28a1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/fvgp/gp.py b/fvgp/gp.py
@@ -110,7 +110,6 @@ def __init__(
         normalize_y = False,
         use_inv = False,
         ram_economy = True,
-        args = None,
         ):
         if input_space_dim != len(x_data[0]):
             raise ValueError("input space dimensions are not in agreement with the point positions given")

diff --git a/tests/test_fvgp.py b/tests/test_fvgp.py
@@ -106,9 +106,9 @@ def test_train_basic(self):
         res = my_gp1.non_stat_kernel(x_data,x_data,np.random.rand(10,5),np.random.rand(10),0.5)
         res = my_gp1.non_stat_kernel_gradient(x_data,x_data,np.random.rand(10,5),np.random.rand(10),0.5)
 
-    def test_train_hgdl_gpu(self):
+    def test_train_hgdl(self):
         my_gp2 = GP(input_dim, x_data,y_data,np.array([1, 1, 1, 1, 1, 1]),variances=np.zeros(y_data.shape) + 0.01,
-            compute_device="gpu", normalize_y = True, use_inv = True, ram_economy = True)
+            compute_device="cpu", normalize_y = True, use_inv = True, ram_economy = True)
 
 
         my_gp2.train(np.array([[0.01,10],[0.01,10],[0.01,10],[0.01,10],[0.01,10],[0.01,10]]),