update

sdaza · Dec 11, 2024 · 9085b5e · 9085b5e
1 parent eb6db31
commit 9085b5e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/experiment_utils/experiment_analyzer.py b/experiment_utils/experiment_analyzer.py
@@ -35,7 +35,7 @@ def __init__(
         instrument_col: str = None,
         alpha: float = 0.05,
         regression_covariates: List = None,
-            assess_overlap=False):
+        assess_overlap=False):
 
         """
         Initialize ExperimentAnalyzer
@@ -176,7 +176,7 @@ def __create_formula(self, outcome_variable, type: str = 'regression'):
         """
 
         formula_dict = {
-            'regression': f"{outcome_variable} ~ {self.treatment_col}",
+            'regression': f"{outcome_variable} ~ 1 + {self.treatment_col}",
             'iv': f"{outcome_variable} ~ 1 + [{self.treatment_col} ~ {self.instrument_col}]"
         }
         reg_covs = list(set(self.final_covariates) & set(self.regression_covariates))