diff --git a/vllm/utils.py b/vllm/utils.py index 65b26ebb9a44f..04e48bfb6b645 100644 --- a/vllm/utils.py +++ b/vllm/utils.py @@ -1657,9 +1657,9 @@ def __len__(self): return len(self._factory) -class ClassRegistry(UserDict[type[T], _V]): +class ClassRegistry(UserDict[Type[T], _V]): - def __getitem__(self, key: type[T]) -> _V: + def __getitem__(self, key: Type[T]) -> _V: for cls in key.mro(): if cls in self.data: return self.data[cls]