Unverified Commit cd4459a1 authored by Nikita Titov's avatar Nikita Titov Committed by GitHub
Browse files

[test][python] fix pandas DeprecationWarning about `is_sparse` function (#6532)

parent 4292b168
......@@ -1812,7 +1812,7 @@ def test_pandas_sparse(rng):
}
)
for dtype in pd.concat([X.dtypes, X_test.dtypes, pd.Series(y.dtypes)]):
assert pd.api.types.is_sparse(dtype)
assert isinstance(dtype, pd.SparseDtype)
params = {"objective": "binary", "verbose": -1}
lgb_train = lgb.Dataset(X, y)
gbm = lgb.train(params, lgb_train, num_boost_round=10)
......
......@@ -638,7 +638,7 @@ def test_pandas_sparse(rng):
}
)
for dtype in pd.concat([X.dtypes, X_test.dtypes, pd.Series(y.dtypes)]):
assert pd.api.types.is_sparse(dtype)
assert isinstance(dtype, pd.SparseDtype)
gbm = lgb.sklearn.LGBMClassifier(n_estimators=10).fit(X, y)
pred_sparse = gbm.predict(X_test, raw_score=True)
if hasattr(X_test, "sparse"):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment