From b98b074a06fcf7aec6f01ed5a0f33e328b399f1c Mon Sep 17 00:00:00 2001 From: AdrianSosic Date: Thu, 5 Sep 2024 22:06:04 +0200 Subject: [PATCH] Use acquisition function abbreviation --- tests/test_iterations.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_iterations.py b/tests/test_iterations.py index bbdff5608..2c9902ed4 100644 --- a/tests/test_iterations.py +++ b/tests/test_iterations.py @@ -4,7 +4,7 @@ import pytest from pytest import param -from baybe.acquisition import qKG, qNIPV, qThompsonSampling, qUCB +from baybe.acquisition import qKG, qNIPV, qTS, qUCB from baybe.acquisition.base import AcquisitionFunction from baybe.exceptions import UnusedObjectWarning from baybe.kernels.base import Kernel @@ -302,7 +302,7 @@ def test_meta_recommenders(campaign, n_iterations, batch_size): run_iterations(campaign, n_iterations, batch_size) -@pytest.mark.parametrize("acqf", [qThompsonSampling(), qUCB()]) +@pytest.mark.parametrize("acqf", [qTS(), qUCB()]) @pytest.mark.parametrize("surrogate_model", [BetaBernoulliMultiArmedBanditSurrogate()]) @pytest.mark.parametrize( "parameter_names",