@@ -81,7 +81,6 @@ def execute_test_suite(self, test_suite: CausalTestSuite) -> list[CausalTestResu
81
81
82
82
estimators = test_suite [edge ]["estimators" ]
83
83
tests = test_suite [edge ]["tests" ]
84
- estimate_type = test_suite [edge ]["estimate_type" ]
85
84
results = {}
86
85
for estimator_class in estimators :
87
86
causal_test_results = []
@@ -96,7 +95,7 @@ def execute_test_suite(self, test_suite: CausalTestSuite) -> list[CausalTestResu
96
95
)
97
96
if estimator .df is None :
98
97
estimator .df = self .scenario_execution_data_df
99
- causal_test_result = self ._return_causal_test_results (estimate_type , estimator , test )
98
+ causal_test_result = self ._return_causal_test_results (estimator , test )
100
99
causal_test_results .append (causal_test_result )
101
100
102
101
results [estimator_class .__name__ ] = causal_test_results
@@ -118,7 +117,6 @@ def execute_test(self, estimator: type(Estimator), causal_test_case: CausalTestC
118
117
119
118
:param estimator: A reference to an Estimator class.
120
119
:param causal_test_case: The CausalTestCase object to be tested
121
- :param estimate_type: A string which denotes the type of estimate to return, ATE or CATE.
122
120
:return causal_test_result: A CausalTestResult for the executed causal test case.
123
121
"""
124
122
if self .scenario_execution_data_df .empty :
@@ -146,7 +144,6 @@ def execute_test(self, estimator: type(Estimator), causal_test_case: CausalTestC
146
144
def _return_causal_test_results (self , estimator , causal_test_case ):
147
145
"""Depending on the estimator used, calculate the 95% confidence intervals and return in a causal_test_result
148
146
149
- :param estimate_type: A string which denotes the type of estimate to return
150
147
:param estimator: An Estimator class object
151
148
:param causal_test_case: The concrete test case to be executed
152
149
:return: a CausalTestResult object containing the confidence intervals
0 commit comments