Fix types

Author: HardNorth

reportportal_client/aio/tasks.py

@@ -13,10 +13,11 @@
 
 """This module contains customized asynchronous Tasks and Task Factories for the ReportPortal client."""
 
+import sys
 import asyncio
 from abc import abstractmethod
 from asyncio import Future
-from typing import Any, Coroutine, Generator, Generic, Optional, TypeVar
+from typing import Any, Coroutine, Generator, Generic, Optional, TypeVar, TypeAlias
 
 # noinspection PyProtectedMember
 from reportportal_client._internal.static.abstract import AbstractBaseClass
@@ -28,6 +29,12 @@ class BlockingOperationError(RuntimeError):
     """An issue with task blocking execution."""
 
 
+if sys.version_info >= (3, 12):
+    _TaskCompatibleCoro: TypeAlias = Coroutine[Any, Any, Any]
+else:
+    _TaskCompatibleCoro: TypeAlias = Generator[Future[object] | None, None, Any] | Coroutine[Any, Any, Any]
+
+
 class Task(Generic[_T], asyncio.Task, metaclass=AbstractBaseClass):
     """Base class for ReportPortal client tasks.
 
@@ -41,7 +48,7 @@ class Task(Generic[_T], asyncio.Task, metaclass=AbstractBaseClass):
 
     def __init__(
         self,
-        coro: Generator[Future[object] | None, None, Any] | Coroutine[Any, Any, Any],
+        coro: _TaskCompatibleCoro,
         *,
         loop: asyncio.AbstractEventLoop,
         name: Optional[str] = None,