From f450da8f265583326e69ffd6e94bfb7a7d7fb255 Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Sun, 14 Jan 2024 21:42:51 +0800 Subject: [PATCH] Fix dask. --- tests/test_distributed/test_gpu_with_dask/test_gpu_with_dask.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_distributed/test_gpu_with_dask/test_gpu_with_dask.py b/tests/test_distributed/test_gpu_with_dask/test_gpu_with_dask.py index a15e1903d238..865623671388 100644 --- a/tests/test_distributed/test_gpu_with_dask/test_gpu_with_dask.py +++ b/tests/test_distributed/test_gpu_with_dask/test_gpu_with_dask.py @@ -363,12 +363,12 @@ def test_early_stopping(self, local_cuda_client: Client) -> None: device="cuda", eval_metric="error", n_estimators=100, + early_stopping_rounds=early_stopping_rounds, ) cls.client = local_cuda_client cls.fit( X, y, - early_stopping_rounds=early_stopping_rounds, eval_set=[(valid_X, valid_y)], ) booster = cls.get_booster()