From 04b2081e9bc20602312be7da23fb302134505bb6 Mon Sep 17 00:00:00 2001 From: CamDavidsonPilon Date: Tue, 29 Oct 2024 06:30:54 -0400 Subject: [PATCH] fix test --- lifelines/tests/test_estimation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lifelines/tests/test_estimation.py b/lifelines/tests/test_estimation.py index abf1d5f97..28087af82 100644 --- a/lifelines/tests/test_estimation.py +++ b/lifelines/tests/test_estimation.py @@ -3320,7 +3320,7 @@ def test_strata_estimation_is_same_if_using_trivial_strata(self, rossi): assert_frame_equal( cph.summary.loc[[("beta_", "Intercept"), ("phi1_", "Intercept")]].reset_index(drop=True), - trivial_strata_cph.summary.loc[[("beta_", "Intercept"), ("sa_phi1_", "Intercept")]].reset_index(drop=True), + trivial_strata_cph.summary.loc[[("beta_", "Intercept"), ("s('a',)_phi1_", "Intercept")]].reset_index(drop=True), atol=0.05, )