Fix type

LQDJudge · Mar 27, 2024 · 28156a9 · 28156a9
1 parent 45469ff
commit 28156a9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/judge/ml/collab_filter.py b/judge/ml/collab_filter.py
@@ -71,7 +71,7 @@ def user_recommendations(self, user_id, problems, measure=DOT, limit=None):
     def problem_neighbors(self, problem, problemset, measure=DOT, limit=None):
         pid = problem.id
         if pid not in self.problem_embeddings:
-            return None
+            return []
         embedding = self.problem_embeddings[pid]
         scores = self.compute_scores(embedding, self.problem_embeddings, measure)
         res = []