fix: add __future__ imports

src/flask_websockets/websockets.py

@@ -1,6 +1,7 @@
+from __future__ import annotations
+
 import contextlib
 import logging
-from collections.abc import Callable, Generator, Iterable
 from functools import wraps
 from typing import TYPE_CHECKING, Any
 
@@ -9,6 +10,7 @@
 from .websocket import WebSocket
 
 if TYPE_CHECKING:
+    from collections.abc import Callable, Generator, Iterable
     from wsgiref.types import StartResponse
 
 
@@ -81,7 +83,7 @@ def websocket_route(*args: Any, **options: Any) -> Response:
                     ws.close()
 
                 class WebSocketResponse(Response):
-                    def __call__(self, environ: dict[str, Any], start_response: "StartResponse") -> Iterable[bytes]:
+                    def __call__(self, environ: dict[str, Any], start_response: StartResponse) -> Iterable[bytes]:
                         if ws.mode == "gunicorn":
                             raise StopIteration()
 

tests/test_client_server_integration.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import platform
 import time
 from queue import Queue

tests/test_flask_sock.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import unittest
 
 from flask import Blueprint, Flask