type hint fixes for adaptive/tests/test_runner.py

Author: basnijholt

adaptive/tests/test_runner.py

@@ -1,9 +1,8 @@
 import asyncio
 import time
-from typing import Callable, Iterator
+from typing import Callable
 
 import pytest
-from distributed.client import Client
 
 from adaptive.learner import BaseLearner, Learner1D, Learner2D
 from adaptive.runner import (
@@ -71,7 +70,7 @@ async def f(x):
 
 
 @pytest.fixture(scope="session")
-def ipyparallel_executor() -> Iterator[Client]:
+def ipyparallel_executor():
     from ipyparallel import Client
     import pexpect
 
@@ -83,7 +82,7 @@ def ipyparallel_executor() -> Iterator[Client]:
 
 
 @pytest.fixture(scope="session")
-def dask_executor() -> Iterator[Client]:
+def dask_executor():
     from distributed import Client
 
     client = Client(n_workers=1)
@@ -114,15 +113,15 @@ def test_stop_after_goal() -> None:
 
 
 @pytest.mark.skipif(not with_ipyparallel, reason="IPyparallel is not installed")
-def test_ipyparallel_executor(ipyparallel_executor: Client) -> None:
+def test_ipyparallel_executor(ipyparallel_executor) -> None:
     learner = Learner1D(linear, (-1, 1))
     BlockingRunner(learner, trivial_goal, executor=ipyparallel_executor)
     assert learner.npoints > 0
 
 
 @pytest.mark.timeout(60)
 @pytest.mark.skipif(not with_distributed, reason="dask.distributed is not installed")
-def test_distributed_executor(dask_executor: Client) -> None:
+def test_distributed_executor(dask_executor) -> None:
     learner = Learner1D(linear, (-1, 1))
     BlockingRunner(learner, trivial_goal, executor=dask_executor)
     assert learner.npoints > 0