fix numpy / torch type bug

lenskit · Jan 6, 2025 · 8472e4b · 8472e4b
1 parent 91633ef
commit 8472e4b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lenskit/lenskit/knn/item.py b/lenskit/lenskit/knn/item.py
@@ -241,7 +241,7 @@ def __call__(self, query: QueryInput, items: ItemList) -> ItemList:
 
         # subset the model to rated and target items
         model = self.sim_matrix_
-        model = model[ri_valid_nums, :]
+        model = model[ri_valid_nums.numpy(), :]
         assert isinstance(model, csr_array)
         model = model[:, ti_valid_nums]
         assert isinstance(model, csr_array)