more to fix test

CamDavidsonPilon · Jan 2, 2024 · f9e778e · f9e778e
1 parent 4fcbff4
commit f9e778e
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/lifelines/fitters/__init__.py b/lifelines/fitters/__init__.py
@@ -550,7 +550,7 @@ def _fit_model(self, Ts, E, entry, weights, fit_options: dict, show_progress=Tru
             minimizing_results, previous_results, minimizing_ll = None, None, np.inf
             for method, option in zip(
                 ["Nelder-Mead", self._scipy_fit_method],
-                [{"maxiter": 200}, {**{"disp": show_progress}, **self._scipy_fit_options, **fit_options}],
+                [{"maxiter": 400}, {**{"disp": show_progress}, **self._scipy_fit_options, **fit_options}],
             ):
 
                 initial_value = self._initial_values if previous_results is None else utils._to_1d_array(previous_results.x)

diff --git a/lifelines/tests/test_estimation.py b/lifelines/tests/test_estimation.py
@@ -5745,6 +5745,4 @@ def _create_initial_point(self, Ts, E, entry, weights):
         T, E = load_kidney_transplant()["time"], load_kidney_transplant()["death"]
         wmc.fit(T, E)
         mcfitter.fit(T, E)
-        print(wmc.summary)
-        print(mcfitter.summary)
         assert_frame_equal(wmc.summary.reset_index(drop=True), mcfitter.summary.reset_index(drop=True), rtol=0.25)