Modified test_woe_encoder_dd.py to fix the string[pyarrow]/object dty…

…pe mismatch
paypal · Mar 5, 2024 · e90f19c · e90f19c
1 parent 9bf4494
commit e90f19c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/gators/encoders/tests/test_woe_encoder_dd.py b/gators/encoders/tests/test_woe_encoder_dd.py
@@ -143,7 +143,8 @@ def test_no_cat_dd_np(data_no_cat):
 def test_data_not_inplace_dd(data_not_inplace):
     obj, X, X_expected = data_not_inplace
     X_new = obj.transform(X).compute()
-    X_new[["A", "B", "C"]] = X_new[["A", "B", "C"]].astype("string[pyarrow]")
+    X_new[["A", "B", "C"]] = X_new[["A", "B", "C"]].astype(object)
+    X_expected[["A", "B", "C"]] = X_expected[["A", "B", "C"]].astype(object)
     X_new[["A__woe", "B__woe", "C__woe"]] = X_new[
         ["A__woe", "B__woe", "C__woe"]
     ].astype(float)