Merge pull request #986 from Sherry-XLL/master

FIX: fix issue #834
RUCAIBox · Oct 7, 2021 · f56bcd0 · f56bcd0
2 parents 434e3f0 + 6069ff8
commit f56bcd0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/recbole/utils/case_study.py b/recbole/utils/case_study.py
@@ -43,7 +43,7 @@ def full_sort_scores(uid_series, model, test_data, device=None):
 
     if not test_data.is_sequential:
         input_interaction = dataset.join(Interaction({uid_field: uid_series}))
-        history_item = test_data.uid2history_item[uid_series]
+        history_item = test_data.uid2history_item[list(uid_series)]
         history_row = torch.cat([torch.full_like(hist_iid, i) for i, hist_iid in enumerate(history_item)])
         history_col = torch.cat(list(history_item))
         history_index = history_row, history_col