Skip to content

Commit 9471f17

Browse files
authored
implicit started using a Matrix type which was breaking this check
1 parent ef84269 commit 9471f17

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_implicit.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,5 +28,5 @@ def test_model(self):
2828
model.fit(counts, show_progress=False)
2929
rows, cols = model.item_factors, model.user_factors
3030

31-
assert not np.isnan(np.sum(cols))
32-
assert not np.isnan(np.sum(rows))
31+
assert not np.isnan(np.sum(cols.to_numpy()))
32+
assert not np.isnan(np.sum(rows.to_numpy()))

0 commit comments

Comments
 (0)