iv estimators with new param format

Author: jmafoster1

causal_testing/testing/causal_test_result.py

@@ -59,9 +59,10 @@ def push(s, inc="  "):
             f"Treatment value: {self.estimator.treatment_value}\n"
             f"Outcome: {self.estimator.outcome}\n"
             f"Adjustment set: {self.adjustment_set}\n"
-            f"Formula: {self.estimator.formula}\n"
-            f"{self.test_value.type}: {result_str}\n"
         )
+        if hasattr(self.estimator, "formula"):
+            base_str += f"Formula: {self.estimator.formula}\n"
+        base_str += f"{self.test_value.type}: {result_str}\n"
         confidence_str = ""
         if self.confidence_intervals:
             ci_str = " " + str(self.confidence_intervals)

causal_testing/testing/estimators.py

@@ -480,8 +480,20 @@ def __init__(
         df: pd.DataFrame = None,
         intercept: int = 1,
         effect_modifiers: dict = None,  # Not used (yet?). Needed for compatibility
+        alpha: float = 0.05,
+        query: str = "",
     ):
-        super().__init__(treatment, treatment_value, control_value, adjustment_set, outcome, df, None)
+        super().__init__(
+            treatment=treatment,
+            treatment_value=treatment_value,
+            control_value=control_value,
+            adjustment_set=adjustment_set,
+            outcome=outcome,
+            df=df,
+            effect_modifiers=None,
+            alpha=alpha,
+            query=query,
+        )
         self.intercept = intercept
         self.model = None
         self.instrument = instrument