Fix python evaluation

It turns out I ran incorrect tests and the evaluation module was not really working
recommenders-team · Jun 26, 2024 · 66ace3e · 66ace3e
1 parent 0a1b98f
commit 66ace3e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/recommenders/evaluation/python_evaluation.py b/recommenders/evaluation/python_evaluation.py
@@ -435,9 +435,9 @@ def merge_ranking_true_pred(
 
     # count the number of hits vs actual relevant items per user
     df_hit_count = pd.merge(
-        df_hit.groupby(col_user, as_index=False)[col_user].agg({"hit": "count"}),
+        df_hit.groupby(col_user, as_index=False)[col_user].agg(hit="count"),
         rating_true_common.groupby(col_user, as_index=False)[col_user].agg(
-            {"actual": "count"}
+            actual="count",
         ),
         on=col_user,
     )