Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Jiaming Yuan 2024-05-14 12:13:35 +08:00 committed by GitHub
parent 871fabeee3
commit f1f69ff10e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/python/test_with_modin.py
View File

@ -1,4 +1,5 @@
import numpy as np
import pandas as pd
import pytest
import xgboost as xgb
@ -71,7 +72,11 @@ class TestModin:
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