Improve type annotations in unified radix cache (#26948)

2026-06-02 21:17:50 +08:00
parent 7271318dc3
commit 38d4c9ba88
3 changed files with 31 additions and 15 deletions
@@ -400,9 +400,9 @@ class HiCacheController:
        self.prefetch_queue = Queue()
        self.backup_queue = Queue()

-        self.prefetch_revoke_queue = Queue()
-        self.ack_backup_queue = Queue()
-        self.host_mem_release_queue = Queue()
+        self.prefetch_revoke_queue: Queue[str] = Queue()
+        self.ack_backup_queue: Queue[StorageOperation] = Queue()
+        self.host_mem_release_queue: Queue[torch.Tensor] = Queue()

        self.prefetch_thread.start()
        self.backup_thread.start()