Cleanup

Signed-off-by: Elena Kolevska <elena@kolevska.com>
dapr · Mar 2, 2024 · a543575 · a543575
1 parent e62ec72
commit a543575
Show file tree

Hide file tree

Showing 5 changed files with 4 additions and 5 deletions.
diff --git a/dapr/clients/grpc/_helpers.py b/dapr/clients/grpc/_helpers.py
@@ -16,7 +16,6 @@
 from enum import Enum
 from google.protobuf.any_pb2 import Any as GrpcAny
 from google.protobuf.message import Message as GrpcMessage
-from grpc import UnaryUnaryClientInterceptor, ClientCallDetails  # type: ignore
 
 MetadataDict = Dict[str, List[Union[bytes, str]]]
 MetadataTuple = Tuple[Tuple[str, Union[bytes, str]], ...]

diff --git a/dapr/clients/grpc/client.py b/dapr/clients/grpc/client.py
@@ -167,7 +167,7 @@ def __init__(
                 options=options,
             )
 
-        self._channel = grpc.intercept_channel(self._channel, DaprClientTimeoutInterceptor())
+        self._channel = grpc.intercept_channel(self._channel, DaprClientTimeoutInterceptor())  # type: ignore
 
         if settings.DAPR_API_TOKEN:
             api_token_interceptor = DaprClientInterceptor(

diff --git a/dapr/clients/grpc/interceptors.py b/dapr/clients/grpc/interceptors.py
@@ -1,7 +1,7 @@
 from collections import namedtuple
 from typing import List, Tuple
 
-from grpc import UnaryUnaryClientInterceptor, ClientCallDetails
+from grpc import UnaryUnaryClientInterceptor, ClientCallDetails  # type: ignore
 
 from dapr.conf import settings
 

diff --git a/tests/clients/test_client_interceptor.py b/tests/clients/test_client_interceptor.py
@@ -15,7 +15,7 @@
 
 import unittest
 
-from dapr.clients.grpc._helpers import DaprClientInterceptor, _ClientCallDetails
+from dapr.clients.grpc.interceptors import DaprClientInterceptor, _ClientCallDetails
 
 
 class DaprClientInterceptorTests(unittest.TestCase):

diff --git a/tests/clients/test_timeout_interceptor.py b/tests/clients/test_timeout_interceptor.py
@@ -15,7 +15,7 @@
 
 import unittest
 from unittest.mock import Mock, patch
-from dapr.clients.grpc.interceptors import DaprClientTimeoutInterceptor, ClientCallDetails
+from dapr.clients.grpc.interceptors import DaprClientTimeoutInterceptor
 from dapr.conf import settings