Reduce rtol in test

secondmind-labs · Aug 1, 2023 · 314fee9 · 314fee9
1 parent ef48ec0
commit 314fee9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unit/models/keras/test_sampler.py b/tests/unit/models/keras/test_sampler.py
@@ -407,7 +407,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=5e-6)
+    npt.assert_allclose(eval_all, tf.concat([eval_1, eval_2, eval_3], axis=0), rtol=1e-5)
 
 
 @random_seed