fix connect_future types in assignment

tornadoweb · Oct 10, 2023 · 33e1dc0 · 33e1dc0
1 parent e492633
commit 33e1dc0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tornado/websocket.py b/tornado/websocket.py
@@ -1360,7 +1360,7 @@ def __init__(
         subprotocols: Optional[List[str]] = None,
         resolver: Optional[Resolver] = None,
     ) -> None:
-        self.connect_future = Future()  # type: Future[WebSocketClientConnection]
+        self.connect_future = Future()  # type: Union[Future[WebSocketClientConnection], None]
         self.read_queue = Queue(1)  # type: Queue[Union[None, str, bytes]]
         self.key = base64.b64encode(os.urandom(16))
         self._on_message_callback = on_message_callback