Fix annotations of chunked function

regal/domain.py

@@ -4,6 +4,7 @@ from enum import IntEnum, unique
 from typing import (
     Any,
     AsyncIterable,
+    Callable,
     Dict,
     List,
     Mapping,
@@ -12,6 +13,7 @@ from typing import (
     Set,
     Type,
     TypedDict,
+    TypeVar,
     Union,
     cast,
     overload,
@@ -20,7 +22,7 @@ from typing import (
 import fred_api.registry.domain.service_admin_grpc_pb2_grpc
 import fred_api.registry.domain.service_domain_grpc_pb2_grpc
 import grpc
-from aioitertools.more_itertools import chunked
+from aioitertools.more_itertools import chunked as _chunked
 from aioitertools.types import AnyIterable
 from backports.strenum import StrEnum
 from fred_api.registry.domain.domain_history_types_pb2 import DomainHistoryReply, DomainHistoryRequest
@@ -55,6 +57,10 @@ from .common import DatetimeRange, ObjectEvents, PaginationResult, RegistryDecod
 from .exceptions import ContactDoesNotExist, DomainDoesNotExist, InvalidHistoryInterval
 from .object import RegistryObjectClient
 
+# XXX: Fix annotation of chunked. Mypy for some reason annotates `chunked` as coroutine and not an async generator.
+T = TypeVar("T")
+chunked = cast(Callable[[AnyIterable[T], int], AsyncIterable[List[T]]], _chunked)
+
 UpdateDomainsNotifyInfoRequest = Union[
     UpdateDomainsDeleteAdditionalNotifyInfoRequest,
     UpdateDomainsOutzoneAdditionalNotifyInfoRequest,