Skip to content

Commit efe57c0

Browse files
Add index for Estimator in test suite results
1 parent d394bdd commit efe57c0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

causal_testing/testing/causal_test_engine.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ def execute_test_suite(self, test_suite: CausalTestSuite) -> list[CausalTestResu
8484
estimators = test_suite_dict[edge]["estimators"]
8585
tests = test_suite_dict[edge]["tests"]
8686
estimate_type = test_suite_dict[edge]["estimate_type"]
87-
results = []
87+
results = {}
8888
for EstimatorClass in estimators:
8989
causal_test_results = []
9090

@@ -104,7 +104,7 @@ def execute_test_suite(self, test_suite: CausalTestSuite) -> list[CausalTestResu
104104
causal_test_result = self._return_causal_test_results(estimate_type, estimator, test)
105105
causal_test_results.append(causal_test_result)
106106

107-
results.append(causal_test_results)
107+
results[EstimatorClass.__name__] = causal_test_results
108108
test_suite_results[edge] = results
109109
return test_suite_results
110110

0 commit comments

Comments
 (0)