Fix SASRec GPU test

chezou · chezou · commit 2f438381fcb3 · 2025-12-04T20:22:38.000-08:00
diff --git a/tests/models/nn/transformers/test_sasrec.py b/tests/models/nn/transformers/test_sasrec.py
@@ -413,7 +413,7 @@ def test_u2i_losses(
             pd.DataFrame(
                 {
                     Columns.User: [30, 30, 30, 40, 40, 40],
-                    Columns.Item: [11, 13, 17, 17, 13, 11],
+                    Columns.Item: [11, 13, 15, 17, 13, 11],
                     Columns.Rank: [1, 2, 3, 1, 2, 3],
                 }
             ),
@@ -454,7 +454,7 @@ def test_u2i_with_key_and_attn_masks(
             pd.DataFrame(
                 {
                     Columns.User: [30, 30, 30, 40, 40, 40],
-                    Columns.Item: [11, 13, 12, 13, 14, 12],
+                    Columns.Item: [11, 12, 13, 13, 14, 12],
                     Columns.Rank: [1, 2, 3, 1, 2, 3],
                 }
             ),

Original file line number	Diff line number	Diff line change
`@@ -413,7 +413,7 @@ def test_u2i_losses(`
`413`	`413`	`pd.DataFrame(`
`414`	`414`	`{`
`415`	`415`	`Columns.User: [30, 30, 30, 40, 40, 40],`
`416`		`- Columns.Item: [11, 13, 17, 17, 13, 11],`
	`416`	`+ Columns.Item: [11, 13, 15, 17, 13, 11],`
`417`	`417`	`Columns.Rank: [1, 2, 3, 1, 2, 3],`
`418`	`418`	`}`
`419`	`419`	`),`
`@@ -454,7 +454,7 @@ def test_u2i_with_key_and_attn_masks(`
`454`	`454`	`pd.DataFrame(`
`455`	`455`	`{`
`456`	`456`	`Columns.User: [30, 30, 30, 40, 40, 40],`
`457`		`- Columns.Item: [11, 13, 12, 13, 14, 12],`
	`457`	`+ Columns.Item: [11, 12, 13, 13, 14, 12],`
`458`	`458`	`Columns.Rank: [1, 2, 3, 1, 2, 3],`
`459`	`459`	`}`
`460`	`460`	`),`