Handle np integer in model slice and prediction. (#10007)

2024-01-26 04:58:48 +08:00
parent a76d6c6131
commit 65d7bf2dfe
7 changed files with 75 additions and 49 deletions
--- a/tests/python/test_predict.py
+++ b/tests/python/test_predict.py
@@ -61,7 +61,7 @@ def run_predict_leaf(device: str) -> np.ndarray:

    validate_leaf_output(leaf, num_parallel_tree)

-    n_iters = 2
+    n_iters = np.int32(2)
    sliced = booster.predict(
        m,
        pred_leaf=True,