Merge branch 'main' into json_read_multiple_files

jmafoster1 · jmafoster1 · commit 066549f97512 · 2023-03-13T10:57:46.000Z
diff --git a/causal_testing/testing/estimators.py b/causal_testing/testing/estimators.py
@@ -155,8 +155,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
`@@ -155,8 +155,7 @@ def _run_logistic_regression(self, data) -> RegressionResultsWrapper:`
`155`	`155`	`treatment_and_adjustments_cols = pd.get_dummies(`
`156`	`156`	`treatment_and_adjustments_cols, columns=[col], drop_first=True`
`157`	`157`	`)`
`158`		`- regression = sm.Logit(outcome_col, treatment_and_adjustments_cols)`
`159`		`- model = regression.fit()`
	`158`	`+ model = sm.Logit(outcome_col, treatment_and_adjustments_cols).fit(disp=0)`
`160`	`159`	`return model`
`161`	`160`
`162`	`161`	`def estimate(self, data: pd.DataFrame) -> RegressionResultsWrapper:`