add type hint on metadata

imkero · imkero · commit a61fb77393e6 · 2025-10-18T04:31:14.000Z
Signed-off-by: Kero Liang &lt;kerorek@outlook.com&gt;
diff --git a/vllm/distributed/device_communicators/shm_object_storage.py b/vllm/distributed/device_communicators/shm_object_storage.py
@@ -157,7 +157,7 @@ def __init__(
 
         if create:
             # we are creating a buffer
-            self.metadata = {}  # monotonic_id -> start address
+            self.metadata: dict[int, int] = {}  # monotonic_id -> start address
             self.shared_memory = shared_memory.SharedMemory(
                 create=True, size=self.data_buffer_size, name=name
             )

Original file line number	Diff line number	Diff line change
`@@ -157,7 +157,7 @@ def __init__(`
`157`	`157`
`158`	`158`	`if create:`
`159`	`159`	`# we are creating a buffer`
`160`		`- self.metadata = {} # monotonic_id -> start address`
	`160`	`+ self.metadata: dict[int, int] = {} # monotonic_id -> start address`
`161`	`161`	`self.shared_memory = shared_memory.SharedMemory(`
`162`	`162`	`create=True, size=self.data_buffer_size, name=name`
`163`	`163`	`)`