Skip to content

Commit 8a9a7b4

Browse files
committed
Fix types
1 parent b5cfafe commit 8a9a7b4

File tree

2 files changed

+10
-5
lines changed

2 files changed

+10
-5
lines changed

reportportal_client/_internal/aio/tasks.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ class BatchedTask(Generic[_T], Task[_T]):
3434

3535
def __init__(
3636
self,
37-
coro: Generator[Future[object] | None, None, Any] | Coroutine[Any, Any, Any],
37+
coro: Union[Generator[Optional[Future[object]], None, Any], Coroutine[Any, Any, Any]],
3838
*,
3939
loop: asyncio.AbstractEventLoop,
4040
name: Optional[str] = None,
@@ -68,7 +68,7 @@ class ThreadedTask(Generic[_T], Task[_T]):
6868

6969
def __init__(
7070
self,
71-
coro: Generator[Future[object] | None, None, Any] | Coroutine[Any, Any, Any],
71+
coro: Union[Generator[Optional[Future[object]], None, Any], Coroutine[Any, Any, Any]],
7272
wait_timeout: float,
7373
*,
7474
loop: asyncio.AbstractEventLoop,

reportportal_client/aio/tasks.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,11 +13,16 @@
1313

1414
"""This module contains customized asynchronous Tasks and Task Factories for the ReportPortal client."""
1515

16-
import sys
1716
import asyncio
17+
import sys
1818
from abc import abstractmethod
1919
from asyncio import Future
20-
from typing import Any, Coroutine, Generator, Generic, Optional, TypeVar, TypeAlias
20+
from typing import Any, Coroutine, Generator, Generic, Optional, TypeVar, Union
21+
22+
if sys.version_info >= (3, 10):
23+
from typing import TypeAlias
24+
else:
25+
from typing_extensions import TypeAlias
2126

2227
# noinspection PyProtectedMember
2328
from reportportal_client._internal.static.abstract import AbstractBaseClass
@@ -32,7 +37,7 @@ class BlockingOperationError(RuntimeError):
3237
if sys.version_info >= (3, 12):
3338
_TaskCompatibleCoro: TypeAlias = Coroutine[Any, Any, Any]
3439
else:
35-
_TaskCompatibleCoro: TypeAlias = Generator[Future[object] | None, None, Any] | Coroutine[Any, Any, Any]
40+
_TaskCompatibleCoro: TypeAlias = Union[Generator[Optional[Future[object]], None, Any], Coroutine[Any, Any, Any]]
3641

3742

3843
class Task(Generic[_T], asyncio.Task, metaclass=AbstractBaseClass):

0 commit comments

Comments
 (0)