mypy.

dmlc · Jan 12, 2021 · c43532b · c43532b
1 parent 273da8e
commit c43532b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python-package/xgboost/sklearn.py b/python-package/xgboost/sklearn.py
@@ -592,7 +592,7 @@ def load_model(self, fname):
 
     def _configure_fit(
         self,
-        booster: Optional[Booster, "XGBModel"],
+        booster: Optional[Union[Booster, "XGBModel"]],
         eval_metric: Optional[Union[Callable, str, List[str]]],
         params: Dict[str, Any],
     ) -> Tuple[Booster, Optional[Metric], Dict[str, Any]]:

diff --git a/tests/python-gpu/test_gpu_with_dask.py b/tests/python-gpu/test_gpu_with_dask.py
@@ -279,7 +279,7 @@ def worker_fn(worker_addr: str, data_ref: Dict) -> None:
                                              workers=[workers[i]]))
             client.gather(futures)
 
-    def test_interface_consistency(self):
+    def test_interface_consistency(self) -> None:
         sig = OrderedDict(signature(dxgb.DaskDMatrix).parameters)
         del sig["client"]
         ddm_names = list(sig.keys())