diff --git a/causalml/metrics/multiqini.py b/causalml/metrics/multiqini.py index c7891020..ae1bd43c 100644 --- a/causalml/metrics/multiqini.py +++ b/causalml/metrics/multiqini.py @@ -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) diff --git a/tests/test_metrics.py b/tests/test_metrics.py index 2b421121..454d4bfe 100644 --- a/tests/test_metrics.py +++ b/tests/test_metrics.py @@ -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)