From 96423ba94e555350e9b30b28b1da4a30eb560f16 Mon Sep 17 00:00:00 2001
From: Riccardo Cappuzzo <riccardo.cappuzzo@gmail.com>
Date: Thu, 5 Dec 2024 16:26:23 +0100
Subject: [PATCH] Fixed test

---
 skrub/tests/test_stringencoder.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/skrub/tests/test_stringencoder.py b/skrub/tests/test_stringencoder.py
index b3d298f19..eb6052bc5 100644
--- 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)