_

rcap107 · rcap107 · commit 995a4d169dc1 · 2026-06-30T12:34:31.000+02:00
diff --git a/skrub/tests/test_session_encoder.py b/skrub/tests/test_session_encoder.py
@@ -555,12 +555,9 @@ def test_add_session_column_old_pandas(df_module):
         }
     )
     session_id = sbd.to_list(
-        sbd.col(
-            _add_session_column(
-                df, [], "timestamp", 30 * 60, session_id_column="timestamp_session_id"
-            ),
-            "timestamp_session_id",
-        )
+        _add_session_column(
+            df, [], "timestamp", 30 * 60, session_id_column="timestamp_session_id"
+        ),
     )
     # Expected: first two events in session 0, last two events in session 1
     assert session_id == [0, 0, 1, 1]

Original file line number	Diff line number	Diff line change
`@@ -555,12 +555,9 @@ def test_add_session_column_old_pandas(df_module):`
`555`	`555`	`}`
`556`	`556`	`)`
`557`	`557`	`session_id = sbd.to_list(`
`558`		`- sbd.col(`
`559`		`- _add_session_column(`
`560`		`- df, [], "timestamp", 30 * 60, session_id_column="timestamp_session_id"`
`561`		`- ),`
`562`		`- "timestamp_session_id",`
`563`		`- )`
	`558`	`+ _add_session_column(`
	`559`	`+ df, [], "timestamp", 30 * 60, session_id_column="timestamp_session_id"`
	`560`	`+ ),`
`564`	`561`	`)`
`565`	`562`	`# Expected: first two events in session 0, last two events in session 1`
`566`	`563`	`assert session_id == [0, 0, 1, 1]`