Fix issues preventing mypy running

Author: eddiebergman

.pre-commit-config.yaml

@@ -39,7 +39,7 @@ repos:
         additional_dependencies: ["toml"] # Needed to parse pyproject.toml
 
   - repo: https://github.com/pre-commit/mirrors-mypy
-    rev: v0.942
+    rev: v0.961
     hooks:
       - id: mypy
         name: mypy auto-sklearn

autosklearn/util/single_thread_client.py

@@ -4,6 +4,7 @@
 from pathlib import Path
 
 import dask.distributed
+from dask.distributed.util import no_default
 
 
 class DummyFuture(dask.distributed.Future):
@@ -18,7 +19,7 @@ def __init__(self, result: typing.Any) -> None:
     def result(self, timeout: typing.Optional[int] = None) -> typing.Any:
         return self._result
 
-    def cancel(self) -> None:
+    def cancel(self, **kwargs) -> None:
         pass
 
     def done(self) -> bool:
@@ -88,7 +89,7 @@ def submit(
         """
         return DummyFuture(func(*args, **kwargs))
 
-    def close(self) -> None:
+    def close(self, timeout=no_default) -> None:
         pass
 
     def shutdown(self) -> None:
@@ -104,7 +105,7 @@ def _get_scheduler_info(self) -> typing.Dict:
             "type": "Scheduler",
         }
 
-    def nthreads(self) -> typing.Dict:
+    def nthreads(self, workers=None, **kwargs) -> typing.Dict:
         return {
             "127.0.0.1": 1,
         }