speed up large model test

lenskit · May 10, 2024 · 5562980 · 5562980
1 parent 3bdb7f7
commit 5562980
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_knn_item_item.py b/tests/test_knn_item_item.py
@@ -340,7 +340,7 @@ def test_ii_large_models(rng):
     _log.info("checking a sample of neighborhoods")
     items = algo_ub.item_index_.values
     items = items[algo_ub.item_counts_.numpy() > 0]
-    for i in rng.choice(items, 100):
+    for i in rng.choice(items, 50):
         ipos = algo_ub.item_index_.get_loc(i)
         _log.debug("checking item %d at position %d", i, ipos)
         assert ipos == algo_lim.item_index_.get_loc(i)