lint

CamDavidsonPilon · Jan 3, 2019 · b1f8d26 · b1f8d26
1 parent 5a4cdb7
commit b1f8d26
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/perf_tests/cp_perf_test.py b/perf_tests/cp_perf_test.py
@@ -11,7 +11,7 @@
     from lifelines.datasets import load_rossi
 
     df = load_rossi()
-    df = pd.concat([df] * 100)
+    df = pd.concat([df] * 20)
     # df['week'] = np.random.exponential(1, size=df.shape[0])
     cp = CoxPHFitter()
     start_time = time.time()

diff --git a/tests/test_estimation.py b/tests/test_estimation.py
@@ -882,7 +882,7 @@ def test_cph_will_handle_times_with_only_censored_individuals(self, rossi):
 
         cph2_summary = CoxPHFitter().fit(rossi, "week", "arrest").summary
 
-        assert cph2_summary['coef'].iloc[0] != cph1_summary['coef'].iloc[0]
+        assert cph2_summary["coef"].iloc[0] != cph1_summary["coef"].iloc[0]
 
     def test_schoenfeld_residuals_no_strata_but_with_censorship(self, cph):
         """