Silenced logistic regression fitting output

jmafoster1 · jmafoster1 · commit 7e7b6ed2c880 · 2023-03-09T13:34:18.000Z
diff --git a/causal_testing/testing/estimators.py b/causal_testing/testing/estimators.py
@@ -152,8 +152,7 @@ def _run_logistic_regression(self, data) -> RegressionResultsWrapper:
                 treatment_and_adjustments_cols = pd.get_dummies(
                     treatment_and_adjustments_cols, columns=[col], drop_first=True
                 )
-        regression = sm.Logit(outcome_col, treatment_and_adjustments_cols)
-        model = regression.fit()
+        model = sm.Logit(outcome_col, treatment_and_adjustments_cols).fit(disp=0)
         return model
 
     def estimate(self, data: pd.DataFrame) -> RegressionResultsWrapper:

Original file line number	Diff line number	Diff line change
`@@ -152,8 +152,7 @@ def _run_logistic_regression(self, data) -> RegressionResultsWrapper:`
`152`	`152`	`treatment_and_adjustments_cols = pd.get_dummies(`
`153`	`153`	`treatment_and_adjustments_cols, columns=[col], drop_first=True`
`154`	`154`	`)`
`155`		`- regression = sm.Logit(outcome_col, treatment_and_adjustments_cols)`
`156`		`- model = regression.fit()`
	`155`	`+ model = sm.Logit(outcome_col, treatment_and_adjustments_cols).fit(disp=0)`
`157`	`156`	`return model`
`158`	`157`
`159`	`158`	`def estimate(self, data: pd.DataFrame) -> RegressionResultsWrapper:`