Skip to content

Commit f1f69ff

Browse files
authored
[CI] Fixes for using the latest modin. (dmlc#10285)
1 parent 871fabe commit f1f69ff

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

tests/python/test_with_modin.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import numpy as np
2+
import pandas as pd
23
import pytest
34

45
import xgboost as xgb
@@ -71,7 +72,11 @@ def test_modin(self) -> None:
7172
np.testing.assert_array_equal(result.columns, exp)
7273
dm = xgb.DMatrix(dummies)
7374
assert dm.feature_names == ['B', 'A_X', 'A_Y', 'A_Z']
74-
assert dm.feature_types == ['int', 'int', 'int', 'int']
75+
if int(pd.__version__[0]) >= 2:
76+
assert dm.feature_types == ["int", "i", "i", "i"]
77+
else:
78+
assert dm.feature_types == ["int", "int", "int", "int"]
79+
7580
assert dm.num_row() == 3
7681
assert dm.num_col() == 4
7782

0 commit comments

Comments
 (0)