Skip to content

Commit 809bccf

Browse files
committed
Fix mypy issues
1 parent 00ebd94 commit 809bccf

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

graphql/pyutils/event_emitter.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from typing import Callable, Dict, List
1+
from typing import cast, Callable, Dict, List, Optional
22

33
from asyncio import AbstractEventLoop, Queue, ensure_future
44
from inspect import isawaitable
@@ -11,7 +11,7 @@
1111
class EventEmitter:
1212
"""A very simple EventEmitter."""
1313

14-
def __init__(self, loop: AbstractEventLoop=None) -> None:
14+
def __init__(self, loop: Optional[AbstractEventLoop]=None) -> None:
1515
self.loop = loop
1616
self.listeners: Dict[str, List[Callable]] = defaultdict(list)
1717

@@ -44,7 +44,8 @@ class EventEmitterAsyncIterator:
4444
"""
4545

4646
def __init__(self, event_emitter: EventEmitter, event_name: str) -> None:
47-
self.queue: Queue = Queue(loop=event_emitter.loop)
47+
self.queue: Queue = Queue(
48+
loop=cast(AbstractEventLoop, event_emitter.loop))
4849
event_emitter.add_listener(event_name, self.queue.put)
4950
self.remove_listener = lambda: event_emitter.remove_listener(
5051
event_name, self.queue.put)

tests/subscription/test_map_async_iterator.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55

66
from graphql.subscription.map_async_iterator import MapAsyncIterator
77

8+
89
async def anext(iterable):
910
"""Return the next item from an async iterator."""
1011
return await iterable.__anext__()

0 commit comments

Comments
 (0)