Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[CI] Workaround mypy errors. (#10754)

Browse Source
This commit is contained in:
Jiaming Yuan
2024-08-27 02:54:11 +08:00
committed by GitHub
parent 25966e4ba8
commit 06c4246ff1
3 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_distributed/test_gpu_with_dask/test_gpu_with_dask.py
View File

@@ -647,10 +647,10 @@ async def run_from_dask_array_asyncio(scheduler_address: str) -> dxgb.TrainRetur
import cupy as cp
X, y, _ = generate_array()
X = X.map_blocks(cp.array) # type: ignore
y = y.map_blocks(cp.array) # type: ignore
X = X.to_backend("cupy")
y = y.to_backend("cupy")
m = await xgb.dask.DaskQuantileDMatrix(client, X, y)
m: xgb.dask.DaskDMatrix = await xgb.dask.DaskQuantileDMatrix(client, X, y) # type: ignore
output = await xgb.dask.train(
client, {"tree_method": "hist", "device": "cuda"}, dtrain=m
)