From ed68fe36885a149759112923d4e7406fcaf62a02 Mon Sep 17 00:00:00 2001 From: Maxim Ziatdinov <34245227+ziatdinovmax@users.noreply.github.com> Date: Sun, 20 Aug 2023 01:26:47 -0400 Subject: [PATCH] fix tests --- tests/test_acq.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_acq.py b/tests/test_acq.py index e933f94..c1e5c39 100644 --- 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))