Skip to content

GitLab

Unverified Commit ffb29862 authored by James Lamb's avatar James Lamb Committed by GitHub
Browse files

[ci] [dask] fix mypy errors about predict() signature (#5827)

parent 6856a45a
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
python-package/lightgbm/dask.py
View file @ ffb29862
......@@ -1235,7 +1235,7 @@ class DaskLGBMClassifier(LGBMClassifier, _DaskLGBMModel):
def predict(
self,
X: _DaskMatrixLike,
X: _DaskMatrixLike, # type: ignore[override]
raw_score: bool = False,
start_iteration: int = 0,
num_iteration: Optional[int] = None,
......@@ -1270,7 +1270,7 @@ class DaskLGBMClassifier(LGBMClassifier, _DaskLGBMModel):
def predict_proba(
self,
X: _DaskMatrixLike,
X: _DaskMatrixLike, # type: ignore[override]
raw_score: bool = False,
start_iteration: int = 0,
num_iteration: Optional[int] = None,
......@@ -1441,7 +1441,7 @@ class DaskLGBMRegressor(LGBMRegressor, _DaskLGBMModel):
def predict(
self,
X: _DaskMatrixLike,
X: _DaskMatrixLike, # type: ignore[override]
raw_score: bool = False,
start_iteration: int = 0,
num_iteration: Optional[int] = None,
......@@ -1616,7 +1616,7 @@ class DaskLGBMRanker(LGBMRanker, _DaskLGBMModel):
def predict(
self,
X: _DaskMatrixLike,
X: _DaskMatrixLike, # type: ignore[override]
raw_score: bool = False,
start_iteration: int = 0,
num_iteration: Optional[int] = None,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment