[CI] Fixes for using the latest modin. (#10285)

dmlc · May 14, 2024 · f1f69ff · f1f69ff
1 parent 871fabe
commit f1f69ff
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/python/test_with_modin.py b/tests/python/test_with_modin.py
@@ -1,4 +1,5 @@
 import numpy as np
+import pandas as pd
 import pytest
 
 import xgboost as xgb
@@ -71,7 +72,11 @@ def test_modin(self) -> None:
         np.testing.assert_array_equal(result.columns, exp)
         dm = xgb.DMatrix(dummies)
         assert dm.feature_names == ['B', 'A_X', 'A_Y', 'A_Z']
-        assert dm.feature_types == ['int', 'int', 'int', 'int']
+        if int(pd.__version__[0]) >= 2:
+            assert dm.feature_types == ["int", "i", "i", "i"]
+        else:
+            assert dm.feature_types == ["int", "int", "int", "int"]
+
         assert dm.num_row() == 3
         assert dm.num_col() == 4