Avoid warning in np primitive type tests. (#7833)

2022-04-23 02:07:01 +08:00
parent c45665a55a
commit 332380479b
3 changed files with 9 additions and 5 deletions
--- a/tests/python/test_basic_models.py
+++ b/tests/python/test_basic_models.py
@@ -569,7 +569,7 @@ class TestModels:
        y = rng.randn(rows)
        feature_names = ["test_feature_" + str(i) for i in range(cols)]
        X_pd = pd.DataFrame(X, columns=feature_names)
-        X_pd.iloc[:, 3] = X_pd.iloc[:, 3].astype(np.int)
+        X_pd.iloc[:, 3] = X_pd.iloc[:, 3].astype(np.int32)

        Xy = xgb.DMatrix(X_pd, y)
        assert Xy.feature_types[3] == "int"