Update some more deprecated type hinting (#17998)

Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
2025-05-13 00:49:33 +01:00
parent acee8f48aa
commit 9d7ea9dbbf
10 changed files with 73 additions and 73 deletions
--- a/vllm/model_executor/custom_op.py
+++ b/vllm/model_executor/custom_op.py
@@ -1,7 +1,5 @@
 # SPDX-License-Identifier: Apache-2.0

-from typing import Dict, Type
-
 import torch.nn as nn

 from vllm.config import get_current_vllm_config
@@ -138,7 +136,7 @@ class CustomOp(nn.Module):
    # Examples:
    # - MyOp.enabled()
    # - op_registry["my_op"].enabled()
-    op_registry: Dict[str, Type['CustomOp']] = {}
+    op_registry: dict[str, type['CustomOp']] = {}

    # Decorator to register custom ops.
    @classmethod