Fixed test

skrub-data · Dec 5, 2024 · 96423ba · 96423ba
1 parent eb4de97
commit 96423ba
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/skrub/tests/test_stringencoder.py b/skrub/tests/test_stringencoder.py
@@ -35,4 +35,8 @@ def test_encoding(encode_column, df_module):
     se = StringEncoder(2)
     result = se.fit_transform(encode_column)
 
+    # Converting dtypes to avoid nullable shenanigans
+    check_df = sbd.pandas_convert_dtypes(check_df)
+    result = sbd.pandas_convert_dtypes(result)
+
     df_module.assert_frame_equal(check_df, result)