Fix dask.

dmlc · Jan 14, 2024 · f450da8 · f450da8
1 parent 560adda
commit f450da8
Showing 1 changed file with 1 addition and 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
@@ -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()