Fix type

2024-03-26 20:23:47 -05:00 · 2024-03-26 20:23:47 -05:00 · 28156a9952
commit 28156a9952
parent 45469ff103
1 changed files with 1 additions and 1 deletions
--- a/judge/ml/collab_filter.py
+++ b/judge/ml/collab_filter.py
@ -71,7 +71,7 @@ class CollabFilter:
    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 = []