fix: Type error due to missing default type-var in zmq.Context (#3307) (

#3308) Co-authored-by: Joongi Kim <[email protected]>
lablup · Dec 26, 2024 · 2cb8b37 · 2cb8b37
1 parent e035a0a
commit 2cb8b37
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/ai/backend/logging/handler/intrinsic.py b/src/ai/backend/logging/handler/intrinsic.py
@@ -19,7 +19,7 @@ def __init__(self, *, endpoint: str, msgpack_options: MsgpackOptions) -> None:
         super().__init__()
         self.endpoint = endpoint
         self.msgpack_options = msgpack_options
-        self._zctx = zmq.Context()
+        self._zctx = zmq.Context[zmq.Socket]()
         # We should use PUSH-PULL socket pairs to avoid
         # lost of synchronization sentinel messages.
         if endpoint:

diff --git a/src/ai/backend/logging/logger.py b/src/ai/backend/logging/logger.py
@@ -309,7 +309,7 @@ def log_worker(
         assert graylog_handler is not None
         graylog_handler.setFormatter(SerializedExceptionFormatter())
 
-    zctx = zmq.Context()
+    zctx = zmq.Context[zmq.Socket]()
     agg_sock = zctx.socket(zmq.PULL)
     agg_sock.bind(log_endpoint)
     ep_url = yarl.URL(log_endpoint)