Skip to content

Commit

Permalink
[CI][WIP] Fixes with latest dependencies.
Browse files Browse the repository at this point in the history
  • Loading branch information
trivialfis committed May 14, 2024
1 parent 871fabe commit 74cfeb0
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion tests/python/test_with_modin.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
import pytest

import xgboost as xgb
import pandas as pd
from xgboost import testing as tm
from xgboost.testing.data import run_base_margin_info

Expand Down Expand Up @@ -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

Expand Down

0 comments on commit 74cfeb0

Please sign in to comment.