add DCG to test

lenskit · Jan 18, 2025 · 7279628 · 7279628
1 parent 1c9dcf5
commit 7279628
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lenskit/tests/eval/test_bulk_metrics.py b/lenskit/tests/eval/test_bulk_metrics.py
@@ -7,7 +7,7 @@
 from lenskit.metrics.basic import ListLength
 from lenskit.metrics.bulk import RunAnalysis
 from lenskit.metrics.predict import RMSE
-from lenskit.metrics.ranking import NDCG, RBP, Precision, RecipRank
+from lenskit.metrics.ranking import DCG, NDCG, RBP, Precision, RecipRank
 from lenskit.testing import demo_recs, ml_ratings
 
 
@@ -34,6 +34,7 @@ def test_recs(demo_recs):
     bms.add_metric(ListLength())
     bms.add_metric(Precision())
     bms.add_metric(NDCG())
+    bms.add_metric(DCG())
     bms.add_metric(RBP)
     bms.add_metric(RecipRank)