Enable ignored checker PYI024

Signed-off-by: yangxuan <[email protected]>
milvus-io · Sep 15, 2023 · b682be8 · b682be8
1 parent 34d7fd1
commit b682be8
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/pymilvus/client/interceptor.py b/pymilvus/client/interceptor.py
@@ -13,8 +13,7 @@
 # limitations under the License.
 """Base class for interceptors that operate on all RPC types."""
 
-import collections
-from typing import Any, Callable, List
+from typing import Any, Callable, List, NamedTuple
 
 import grpc
 
@@ -73,10 +72,14 @@ def intercept_stream_stream(
         return postprocess(response_it) if postprocess else response_it
 
 
-class _ClientCallDetails(
-    collections.namedtuple("_ClientCallDetails", ("method", "timeout", "metadata", "credentials")),
-    grpc.ClientCallDetails,
-):
+class ClientCallDetailsTuple(NamedTuple):
+    method: Any
+    timeout: Any
+    metadata: Any
+    credentials: Any
+
+
+class _ClientCallDetails(ClientCallDetailsTuple, grpc.ClientCallDetails):
     pass
 
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -92,7 +92,6 @@ ignore = [
     "TRY400",
     "PLR0912", # TODO
     "C901", # TODO
-    "PYI024", # TODO
     "PYI041", # TODO
 ]