Skip to content

Commit 3a6e2df

Browse files
committed
Fixed logistic regression test
1 parent 31bf002 commit 3a6e2df

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/estimation_tests/test_logistic_regression_estimator.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,5 +19,5 @@ def setUpClass(cls) -> None:
1919
def test_odds_ratio(self):
2020
df = self.scarf_df.copy()
2121
logistic_regression_estimator = LogisticRegressionEstimator("length_in", 65, 55, set(), "completed", df)
22-
odds = logistic_regression_estimator.estimate_unit_odds_ratio()
23-
self.assertEqual(round(odds, 4), 0.8948)
22+
odds, _ = logistic_regression_estimator.estimate_unit_odds_ratio()
23+
self.assertEqual(round(odds[0], 4), 0.8948)

0 commit comments

Comments
 (0)