Skip to content

Commit 9c7e4cd

Browse files
committed
Records the formula actually used in estimation reguardless of whether one was supplied
1 parent b4d3030 commit 9c7e4cd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

causal_testing/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -369,7 +369,7 @@ def save_results(self, results: List[CausalTestResult]) -> None:
369369
"effect": test_config.get("effect", "direct"),
370370
"treatment_variable": test_config["treatment_variable"],
371371
"expected_effect": test_config["expected_effect"],
372-
"formula": test_config.get("formula"),
372+
"formula": result.estimator.formula if hasattr(result.estimator, "formula") else None,
373373
"alpha": test_config.get("alpha", 0.05),
374374
"skip": test_config.get("skip", False),
375375
"passed": test_passed,

0 commit comments

Comments
 (0)