From 5a9e56056f32519f278a46ad5ceff1b44324e26f Mon Sep 17 00:00:00 2001 From: Zichen Wang Date: Tue, 5 Dec 2023 23:11:53 +0000 Subject: [PATCH] Fix mrr unit test --- tests/unit-tests/test_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit-tests/test_utils.py b/tests/unit-tests/test_utils.py index 585f5bbd57..11498230d6 100644 --- a/tests/unit-tests/test_utils.py +++ b/tests/unit-tests/test_utils.py @@ -774,7 +774,7 @@ def test_gen_mrr_score(): metrics_opti = gen_mrr_score(ranking) - assert th.isclose(metrics['mrr'], metrics_opti['mrr']) # Default tolerance: 1e-08 + assert th.isclose(metrics['mrr'], metrics_opti) # Default tolerance: 1e-08 def test_stream_dist_tensors_to_hdf5(): with tempfile.TemporaryDirectory() as tmpdirname: