bug fix in gp.py

lbl-camera · Mar 25, 2024 · 7670e5e · 7670e5e
1 parent 3caa788
commit 7670e5e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/fvgp/gp.py b/fvgp/gp.py
@@ -1265,6 +1265,7 @@ def _LU(self, KV, vec):
         return KVinvY, KVlogdet, factorization_obj
 
     def _Chol(self, KV, vec):
+        st = time.time()
         if self.info: print("Dense Cholesky in progress ...")
         c, l = cho_factor(KV)
         factorization_obj = ("Chol", c, l)

diff --git a/tests/test_fvgp.py b/tests/test_fvgp.py
@@ -40,10 +40,10 @@
 
 """Tests for `fvgp` package."""
 def test_single_task_init_basic():
-    my_gp1 = GP(input_dim, x_data, y_data, init_hyperparameters = np.array([1, 1, 1, 1, 1, 1]), compute_device = 'cpu')
-    my_gp1 = GP(input_dim, x_data, y_data)
+    my_gp1 = GP(input_dim, x_data, y_data, init_hyperparameters = np.array([1, 1, 1, 1, 1, 1]), compute_device = 'cpu', info = True)
+    my_gp1 = GP(input_dim, x_data, y_data, info = True)
     my_gp1 = GP(input_dim, x_data, y_data, init_hyperparameters = np.array([1, 1, 1, 1, 1, 1]))
-    my_gp1 = GP(input_dim, x_data, y_data, init_hyperparameters = np.array([1, 1, 1, 1, 1, 1]), store_inv = False)
+    my_gp1 = GP(input_dim, x_data, y_data, init_hyperparameters = np.array([1, 1, 1, 1, 1, 1]), store_inv = False, info = True)
     my_gp1 = GP(input_dim, x_data, y_data, init_hyperparameters = np.array([1, 1, 1, 1, 1, 1]), args = {'a':2.})