|
14 | 14 | from uuid import UUID |
15 | 15 | import pytest |
16 | 16 | from contextlib import contextmanager |
17 | | -from typing import Callable, Iterator, Set |
| 17 | +from typing import Callable, Iterator, Set, Union |
18 | 18 | from tenacity import Retrying |
19 | 19 |
|
20 | 20 | DEFAULT_TIMEOUT_SECONDS = 15 * 60 # 15 min |
|
24 | 24 | def create_sleeper_jobs( |
25 | 25 | api_client: osparc.ApiClient, |
26 | 26 | sleeper: osparc.Solver, |
27 | | -) -> Callable[[int], Iterator[Set[UUID | str]]]: |
| 27 | +) -> Callable[[int], Iterator[Set[Union[UUID, str]]]]: |
28 | 28 | @contextmanager |
29 | | - def sleeper_jobs(n_jobs: int = 1) -> Iterator[Set[UUID | str]]: |
| 29 | + def sleeper_jobs(n_jobs: int = 1) -> Iterator[Set[Union[UUID, str]]]: |
30 | 30 | job_ids = set() |
31 | 31 | solvers_api = osparc.SolversApi(api_client=api_client) |
32 | 32 | try: |
@@ -55,7 +55,7 @@ def sleeper_jobs(n_jobs: int = 1) -> Iterator[Set[UUID | str]]: |
55 | 55 | @skip_if_osparc_version(at_least=Version("0.8.3.post0.dev20")) |
56 | 56 | def test_jobs( |
57 | 57 | api_client: osparc.ApiClient, |
58 | | - create_sleeper_jobs: Callable[[int], Iterator[Set[UUID | str]]], |
| 58 | + create_sleeper_jobs: Callable[[int], Iterator[Set[Union[UUID, str]]]], |
59 | 59 | sleeper: osparc.Solver, |
60 | 60 | ): |
61 | 61 | """Test the jobs method |
@@ -90,7 +90,7 @@ def test_jobs( |
90 | 90 | async def test_logstreaming( |
91 | 91 | api_client: osparc.ApiClient, |
92 | 92 | sleeper: osparc.Solver, |
93 | | - create_sleeper_jobs: Callable[[int], Iterator[Set[UUID]]], |
| 93 | + create_sleeper_jobs: Callable[[int], Iterator[Set[Union[UUID, str]]]], |
94 | 94 | async_client: AsyncClient, |
95 | 95 | ): |
96 | 96 | """Test log streaming""" |
|
0 commit comments