Add __slots__ to multi callables

XuanWang-Amos · Jan 3, 2024 · 0f655a5 · 0f655a5
1 parent b023ae1
commit 0f655a5
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/src/python/grpcio/grpc/_channel.py b/src/python/grpcio/grpc/_channel.py
@@ -1055,6 +1055,9 @@ class _UnaryUnaryMultiCallable(grpc.UnaryUnaryMultiCallable):
     _response_deserializer: Optional[DeserializingFunction]
     _context: Any
 
+    __slots__ = ["_channel", "_managed_call", "_method", "_target",
+                 "_request_serializer", "_response_deserializer", "_context"]
+
     # pylint: disable=too-many-arguments
     def __init__(
         self,
@@ -1225,6 +1228,9 @@ class _SingleThreadedUnaryStreamMultiCallable(grpc.UnaryStreamMultiCallable):
     _response_deserializer: Optional[DeserializingFunction]
     _context: Any
 
+    __slots__ = ["_channel", "_method", "_target",
+                 "_request_serializer", "_response_deserializer", "_context"]
+
     # pylint: disable=too-many-arguments
     def __init__(
         self,
@@ -1313,6 +1319,9 @@ class _UnaryStreamMultiCallable(grpc.UnaryStreamMultiCallable):
     _response_deserializer: Optional[DeserializingFunction]
     _context: Any
 
+    __slots__ = ["_channel", "_managed_call", "_method", "_target",
+                 "_request_serializer", "_response_deserializer", "_context"]
+
     # pylint: disable=too-many-arguments
     def __init__(
         self,
@@ -1394,6 +1403,9 @@ class _StreamUnaryMultiCallable(grpc.StreamUnaryMultiCallable):
     _response_deserializer: Optional[DeserializingFunction]
     _context: Any
 
+    __slots__ = ["_channel", "_managed_call", "_method", "_target",
+                 "_request_serializer", "_response_deserializer", "_context"]
+
     # pylint: disable=too-many-arguments
     def __init__(
         self,
@@ -1555,6 +1567,9 @@ class _StreamStreamMultiCallable(grpc.StreamStreamMultiCallable):
     _response_deserializer: Optional[DeserializingFunction]
     _context: Any
 
+    __slots__ = ["_channel", "_managed_call", "_method", "_target",
+                 "_request_serializer", "_response_deserializer", "_context"]
+
     # pylint: disable=too-many-arguments
     def __init__(
         self,