Skip to content

Commit

Permalink
lint
Browse files Browse the repository at this point in the history
  • Loading branch information
rolandrmgservices committed Dec 11, 2023
1 parent 2ee63b7 commit 575f1a4
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 1 deletion.
1 change: 1 addition & 0 deletions causalml/metrics/multiqini.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@ def get_mq_ipw_scores(Y, W, W_hat=None):
"""Construct evaluation scores via inverse-propensity weighting.
See https://github.com/grf-labs/maq/blob/master/python-package/maq/maq.py
"""

return get_ipw_scores(Y, W, W_hat)


Expand Down
2 changes: 1 addition & 1 deletion tests/test_metrics.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,4 +41,4 @@ def test_multiqini():
mq.fit(tau_hat, cost, DR_scores)

# (0.005729002695991717, 0.019814651108894354)
assert isclose(mq.average_gain(spend=0.1)[0],0.005729)
assert isclose(mq.average_gain(spend=0.1)[0], 0.005729)

0 comments on commit 575f1a4

Please sign in to comment.