Update function signature

airtai · May 24, 2024 · 64b0704 · 64b0704
1 parent d76b1b2
commit 64b0704
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/faststream/confluent/client.py b/faststream/confluent/client.py
@@ -3,7 +3,6 @@
 from ssl import SSLContext
 from time import time
 from typing import (
-    TYPE_CHECKING,
     Any,
     Callable,
     Dict,
@@ -23,9 +22,6 @@
 from faststream.log import logger
 from faststream.utils.functions import call_or_await
 
-if TYPE_CHECKING:
-    from faststream.types import LoggerProto
-
 _missing = object()
 
 
@@ -112,7 +108,7 @@ def __init__(
         sasl_kerberos_domain_name: Optional[str] = None,
         sasl_oauth_token_provider: Optional[str] = None,
         logger: Annotated[
-            Union["LoggerProto", logging.Logger, None, object],
+            Union[logging.Logger, None, object],
             Doc("User specified logger to pass into Context and log service messages."),
         ] = logger,
     ) -> None:
@@ -307,7 +303,7 @@ def __init__(
         sasl_kerberos_domain_name: Optional[str] = None,
         sasl_oauth_token_provider: Optional[str] = None,
         logger: Annotated[
-            Union["LoggerProto", logging.Logger, None, object],
+            Union[logging.Logger, None, object],
             Doc("User specified logger to pass into Context and log service messages."),
         ] = logger,
     ) -> None: