resolve ci test falling

teodor-r · teodor-r · commit 50358589ceae · 2025-07-24T18:34:25.000+03:00
diff --git a/tests/models/nn/transformers/test_sasrec.py b/tests/models/nn/transformers/test_sasrec.py
@@ -412,7 +412,7 @@ def test_u2i_losses(
             pd.DataFrame(
                 {
                     Columns.User: [30, 30, 30, 40, 40, 40],
-                    Columns.Item: [11, 13, 15, 17, 13, 11],
+                    Columns.Item: [11, 13, 17, 17, 13, 11],
                     Columns.Rank: [1, 2, 3, 1, 2, 3],
                 }
             ),
@@ -453,7 +453,7 @@ def test_u2i_with_key_and_attn_masks(
             pd.DataFrame(
                 {
                     Columns.User: [30, 30, 30, 40, 40, 40],
-                    Columns.Item: [11, 12, 13, 13, 14, 12],
+                    Columns.Item: [11, 13, 12, 13, 14, 12],
                     Columns.Rank: [1, 2, 3, 1, 2, 3],
                 }
             ),

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