Use acquisition function abbreviation

emdgroup · Sep 5, 2024 · b98b074 · b98b074
1 parent d60c3a3
commit b98b074
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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",