Try & catch type error for different risk_ratio implementations

Author: christopher-wild

causal_testing/testing/causal_test_engine.py

@@ -162,7 +162,10 @@ def _return_causal_test_results(self, estimator, causal_test_case):
             )
         elif causal_test_case.estimate_type == "risk_ratio":
             logger.debug("calculating risk_ratio")
-            risk_ratio, confidence_intervals = estimator.estimate_risk_ratio(causal_test_case.estimate_params)
+            try:
+                risk_ratio, confidence_intervals = estimator.estimate_risk_ratio(causal_test_case.estimate_params)
+            except TypeError:
+                risk_ratio, confidence_intervals = estimator.estimate_risk_ratio()
             causal_test_result = CausalTestResult(
                 estimator=estimator,
                 test_value=TestValue("risk_ratio", risk_ratio),

causal_testing/testing/estimators.py

@@ -360,7 +360,7 @@ def estimate_unit_ate(self) -> float:
             ci_high = ci_high[0]
         return unit_effect, [ci_low, ci_high]
 
-    def estimate_ate(self, estimator_params: dict = None) -> tuple[float, list[float, float], float]:
+    def estimate_ate(self) -> tuple[float, list[float, float], float]:
         """Estimate the average treatment effect of the treatment on the outcome. That is, the change in outcome caused
         by changing the treatment variable from the control value to the treatment value.
 
@@ -411,7 +411,7 @@ def estimate_control_treatment(self, adjustment_config: dict = None) -> tuple[pd
 
         return y.iloc[1], y.iloc[0]
 
-    def estimate_risk_ratio(self, estimator_params: dict = None) -> tuple[float, list[float, float]]:
+    def estimate_risk_ratio(self) -> tuple[float, list[float, float]]:
         """Estimate the risk_ratio effect of the treatment on the outcome. That is, the change in outcome caused
         by changing the treatment variable from the control value to the treatment value.