fix tests

ziatdinovmax · Aug 20, 2023 · ed68fe3 · ed68fe3
1 parent 7ba47c3
commit ed68fe3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_acq.py b/tests/test_acq.py
@@ -49,8 +49,8 @@ def test_UCB_beta():
     y = 10 * X**2
     m = ExactGP(1, 'RBF')
     m.fit(rng_keys[0], X, y, num_warmup=100, num_samples=100)
-    obj1 = UCB(rng_keys[1], m, X_new, 2)
-    obj2 = UCB(rng_keys[1], m, X_new, 2)
+    obj1 = UCB(rng_keys[1], m, X_new, beta=2)
+    obj2 = UCB(rng_keys[1], m, X_new, beta=4)
     assert_(not onp.array_equal(obj1, obj2))