diff --git a/tests/studies/test_risk_studies.py b/tests/studies/test_risk_studies.py index b2c4a270..56df2c95 100644 --- a/tests/studies/test_risk_studies.py +++ b/tests/studies/test_risk_studies.py @@ -78,7 +78,8 @@ def test_surv_search(sample_for_search: bool) -> None: metrics = evaluate_survival_estimator(model_v2, X, T, Y, eval_time_horizons) score_v2 = metrics["raw"]["c_index"][0] - assert score_v2 >= score_v1 + EPS = 0.05 + assert score_v2 + EPS >= score_v1 model = study.fit() assert model.is_fitted()