Fix mrr unit test

awslabs · Dec 5, 2023 · 5a9e560 · 5a9e560
1 parent 38d6203
commit 5a9e560
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: