Fix type hint of OnMessageCallback (Firehose client) (#183)

MarshalX · Nov 6, 2023 · dacea8f · dacea8f
1 parent b19763c
commit dacea8f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/atproto/firehose/client.py b/atproto/firehose/client.py
@@ -27,7 +27,7 @@
 _BASE_WEBSOCKET_URI = 'wss://bsky.social/xrpc'
 _MAX_MESSAGE_SIZE_BYTES = 1024 * 1024 * 5  # 5MB
 
-OnMessageCallback = t.Callable[['MessageFrame'], t.Generator[t.Any, None, t.Any]]
+OnMessageCallback = t.Callable[['MessageFrame'], None]
 AsyncOnMessageCallback = t.Callable[['MessageFrame'], t.Coroutine[t.Any, t.Any, None]]
 
 OnCallbackErrorCallback = t.Callable[[BaseException], None]