fix test

CamDavidsonPilon · Oct 29, 2024 · 04b2081 · 04b2081
1 parent ef27139
commit 04b2081
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,
         )