[CI] Workaround mypy errors. (#10754)

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

View File

@@ -6,8 +6,7 @@ dependencies:
- pylint
- wheel
- setuptools
# https://github.com/python/mypy/issues/17617
- mypy !=1.11.0,!=1.11.1
- mypy
- numpy
- scipy
- pandas

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
)