Skip to content

Commit

Permalink
fix outstanding test on Andrin's branch
Browse files Browse the repository at this point in the history
  • Loading branch information
EgorKraevTransferwise committed Nov 27, 2024
1 parent c46560e commit 04342ef
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/causaltune/test_erupt.py
Original file line number Diff line number Diff line change
Expand Up @@ -79,7 +79,7 @@ def call_erupt(self, df: pd.DataFrame, target_score: float, X_names: List[str]):
e = ERUPT("treatment", DummyClassifier(strategy="prior"), X_names=X_names)
e.fit(df)
out = e.score(df, outcome=df["outcome"], policy=df["policy"])
assert out == target_score, "Wrong binary ERUPT score"
assert out == pytest.approx(target_score, rel=1e-5), "Wrong binary ERUPT score"


if __name__ == "__main__":
Expand Down

0 comments on commit 04342ef

Please sign in to comment.