Follow-up on f42fd7b.

python-websockets · Jan 18, 2023 · 7162452 · 7162452
1 parent f42fd7b
commit 7162452
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/src/websockets/legacy/auth.py b/src/websockets/legacy/auth.py
@@ -119,7 +119,7 @@ def basic_auth_protocol_factory(
     credentials: Optional[Union[Credentials, Iterable[Credentials]]] = None,
     check_credentials: Optional[Callable[[str, str], Awaitable[bool]]] = None,
     create_protocol: Optional[Callable[..., BasicAuthWebSocketServerProtocol]] = None,
-) -> Callable[[Any], BasicAuthWebSocketServerProtocol]:
+) -> Callable[..., BasicAuthWebSocketServerProtocol]:
     """
     Protocol factory that enforces HTTP Basic Auth.
 
@@ -175,11 +175,7 @@ async def check_credentials(username: str, password: str) -> bool:
             return hmac.compare_digest(expected_password, password)
 
     if create_protocol is None:
-        # Not sure why mypy cannot figure this out.
-        create_protocol = cast(
-            Callable[[Any], BasicAuthWebSocketServerProtocol],
-            BasicAuthWebSocketServerProtocol,
-        )
+        create_protocol = BasicAuthWebSocketServerProtocol
 
     return functools.partial(
         create_protocol,