Skip to content

Commit 48685ae

Browse files
committed
Fix implicit BPR test with GPU
1 parent d08ef45 commit 48685ae

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/models/test_implicit_bpr.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -98,9 +98,9 @@ def dataset(self) -> Dataset:
9898
),
9999
pd.DataFrame(
100100
{
101-
Columns.User: [10, 10, 20, 20],
102-
Columns.Item: [17, 15, 17, 15],
103-
Columns.Rank: [1, 2, 1, 2],
101+
Columns.User: [20, 20],
102+
Columns.Item: [17, 15],
103+
Columns.Rank: [1, 2],
104104
}
105105
),
106106
),

0 commit comments

Comments
 (0)