From bb9429329fdfbd5896658f41ac25aa2b13d5345d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sebastian=20P=C3=B6lsterl?= Date: Sun, 16 Feb 2025 17:40:51 +0100 Subject: [PATCH] Fix error message --- sksurv/nonparametric.py | 2 +- tests/test_nonparametric.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/sksurv/nonparametric.py b/sksurv/nonparametric.py index 5d0d1f81..71cc9630 100644 --- a/sksurv/nonparametric.py +++ b/sksurv/nonparametric.py @@ -745,7 +745,7 @@ def cumulative_incidence_competing_risks( elif var_type == "Aalen": var = _var_aalen(n_events_cr, kpe_prime, n_at_risk, cum_inc) else: - raise ValueError(f"{var_type=} must be one of 'Dinse', 'Dinse_approx' or 'Aalen'.") + raise ValueError(f"{var_type=} must be one of 'Dinse', 'Dinse_Approx' or 'Aalen'.") _x, _y, conf_int_km = kaplan_meier_estimator(event > 0, time_exit, conf_type="log-log") ci = np.empty(shape=(2, n_risks + 1, n_t), dtype=conf_int_km.dtype) diff --git a/tests/test_nonparametric.py b/tests/test_nonparametric.py index 4b661c0a..4519e00a 100644 --- a/tests/test_nonparametric.py +++ b/tests/test_nonparametric.py @@ -7657,6 +7657,6 @@ def test_invalid_conf_type_competing_risks(event, time, true_x, true_y, conf_typ @pytest.mark.parametrize("event, time, true_x, true_y", SimpleDataBMTCases().get_cases()) @pytest.mark.parametrize("var_type", ["None", "dinse", 1, "", "not"]) def test_invalid_var_type_competing_risks(event, time, true_x, true_y, var_type): - msg = f"{var_type=} must be one of 'Dinse', 'Dinse_approx' or 'Aalen'." + msg = f"{var_type=} must be one of 'Dinse', 'Dinse_Approx' or 'Aalen'." with pytest.raises(ValueError, match=msg): cumulative_incidence_competing_risks(event, time, conf_level=0.95, conf_type="log-log", var_type=var_type)