Fix alltests (#563)

secondmind-labs · Jun 7, 2022 · acc52e4 · acc52e4
1 parent d453976
commit acc52e4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/tests/integration/test_bayesian_optimization.py b/tests/integration/test_bayesian_optimization.py
@@ -327,17 +327,17 @@ def test_bayesian_optimizer_with_sgpr_finds_minima_of_simple_quadratic() -> None
 @pytest.mark.parametrize(
     "num_steps, acquisition_rule",
     [
-        pytest.param(10, DiscreteThompsonSampling(1000, 2), id="DiscreteThompsonSampling"),
+        pytest.param(25, DiscreteThompsonSampling(1000, 8), id="DiscreteThompsonSampling"),
         pytest.param(
-            10,
+            25,
             EfficientGlobalOptimization(
                 ParallelContinuousThompsonSampling(),
                 num_query_points=3,
             ),
             id="ParallelContinuousThompsonSampling",
         ),
         pytest.param(
-            6,
+            12,
             EfficientGlobalOptimization(
                 GreedyContinuousThompsonSampling(),
                 num_query_points=4,
@@ -406,9 +406,9 @@ def test_bayesian_optimizer_with_dgp_finds_minima_of_simple_quadratic(
 @pytest.mark.parametrize(
     "num_steps, acquisition_rule",
     [
-        pytest.param(35, EfficientGlobalOptimization(), id="EfficientGlobalOptimization"),
+        pytest.param(60, EfficientGlobalOptimization(), id="EfficientGlobalOptimization"),
         pytest.param(
-            20,
+            30,
             EfficientGlobalOptimization(
                 ParallelContinuousThompsonSampling(),
                 num_query_points=4,

diff --git a/tests/unit/models/keras/test_sampler.py b/tests/unit/models/keras/test_sampler.py
@@ -319,7 +319,7 @@ def test_ensemble_trajectory_sampler_trajectory_on_subsets_same_as_set(diversify
     eval_2 = trajectory(test_data[100:200, :])
     eval_3 = trajectory(test_data[200:300, :])
 
-    npt.assert_allclose(eval_all, tf.concat([eval_1, eval_2, eval_3], axis=0), rtol=1e-6)
+    npt.assert_allclose(eval_all, tf.concat([eval_1, eval_2, eval_3], axis=0), rtol=2e-6)
 
 
 @random_seed