Skip to content

Commit 59a9fdb

Browse files
committed
Updated doc strings and missed method call
1 parent bf0c3c5 commit 59a9fdb

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

causal_testing/testing/causal_test_engine.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,6 @@ def execute_test_suite(self, test_suite: CausalTestSuite) -> list[CausalTestResu
8181

8282
estimators = test_suite[edge]["estimators"]
8383
tests = test_suite[edge]["tests"]
84-
estimate_type = test_suite[edge]["estimate_type"]
8584
results = {}
8685
for estimator_class in estimators:
8786
causal_test_results = []
@@ -96,7 +95,7 @@ def execute_test_suite(self, test_suite: CausalTestSuite) -> list[CausalTestResu
9695
)
9796
if estimator.df is None:
9897
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)
10099
causal_test_results.append(causal_test_result)
101100

102101
results[estimator_class.__name__] = causal_test_results
@@ -118,7 +117,6 @@ def execute_test(self, estimator: type(Estimator), causal_test_case: CausalTestC
118117
119118
:param estimator: A reference to an Estimator class.
120119
: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.
122120
:return causal_test_result: A CausalTestResult for the executed causal test case.
123121
"""
124122
if self.scenario_execution_data_df.empty:
@@ -146,7 +144,6 @@ def execute_test(self, estimator: type(Estimator), causal_test_case: CausalTestC
146144
def _return_causal_test_results(self, estimator, causal_test_case):
147145
"""Depending on the estimator used, calculate the 95% confidence intervals and return in a causal_test_result
148146
149-
:param estimate_type: A string which denotes the type of estimate to return
150147
:param estimator: An Estimator class object
151148
:param causal_test_case: The concrete test case to be executed
152149
:return: a CausalTestResult object containing the confidence intervals

0 commit comments

Comments
 (0)