Skip to content

Commit 2d68658

Browse files
Add optional estimator type
1 parent a5d6cff commit 2d68658

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

causal_testing/testing/causal_test_suite.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ def __init__(
1313
):
1414
self.test_suite = {}
1515

16-
def add_test_object(self, base_test_case, causal_test_case_list, estimators, estimate_type):
16+
def add_test_object(self, base_test_case, causal_test_case_list, estimators, estimate_type: str = "ate"):
1717
test_object = {"tests": causal_test_case_list, "estimators": estimators, "estimate_type": estimate_type}
1818
self.test_suite[base_test_case] = test_object
1919

0 commit comments

Comments
 (0)