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

4
python-package/xgboost/dask/__init__.py
View File

@@ -1634,7 +1634,7 @@ class DaskScikitLearnBase(XGBModel):
if isinstance(predts, dd.DataFrame):
predts = predts.to_dask_array()
else:
test_dmatrix = await DaskDMatrix(
test_dmatrix: DaskDMatrix = await DaskDMatrix( # type: ignore
self.client,
data=data,
base_margin=base_margin,
@@ -1675,7 +1675,7 @@ class DaskScikitLearnBase(XGBModel):
iteration_range: Optional[IterationRange] = None,
) -> Any:
iteration_range = self._get_iteration_range(iteration_range)
test_dmatrix = await DaskDMatrix(
test_dmatrix: DaskDMatrix = await DaskDMatrix( # type: ignore
self.client,
data=X,
missing=self.missing,