Skip to content

Commit 5562980

Browse files
committed
speed up large model test
1 parent 3bdb7f7 commit 5562980

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_knn_item_item.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -340,7 +340,7 @@ def test_ii_large_models(rng):
340340
_log.info("checking a sample of neighborhoods")
341341
items = algo_ub.item_index_.values
342342
items = items[algo_ub.item_counts_.numpy() > 0]
343-
for i in rng.choice(items, 100):
343+
for i in rng.choice(items, 50):
344344
ipos = algo_ub.item_index_.get_loc(i)
345345
_log.debug("checking item %d at position %d", i, ipos)
346346
assert ipos == algo_lim.item_index_.get_loc(i)

0 commit comments

Comments
 (0)