Add batch TREGO integ test

secondmind-labs · Aug 23, 2023 · f8b46e9 · f8b46e9
1 parent 8545a84
commit f8b46e9
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/integration/test_bayesian_optimization.py b/tests/integration/test_bayesian_optimization.py
@@ -200,6 +200,17 @@ def GPR_OPTIMIZER_PARAMS() -> Tuple[str, List[ParameterSet]]:
                 BatchTrustRegionBox([TREGOBox(ScaledBranin.search_space)]),
                 id="TREGO",
             ),
+            pytest.param(
+                20,
+                BatchTrustRegionBox(
+                    [TREGOBox(ScaledBranin.search_space) for _ in range(3)],
+                    EfficientGlobalOptimization(
+                        ParallelContinuousThompsonSampling(),
+                        num_query_points=3,
+                    ),
+                ),
+                id="TREGO/ParallelContinuousThompsonSampling",
+            ),
             pytest.param(
                 15,
                 BatchTrustRegionBox(