[Build][Bugfix] Using the correct type hint (vllm-project#10866)

Signed-off-by: Gregory Shtrasberg <Gregory.Shtrasberg@amd.com>
vanbasten23 · Dec 3, 2024 · a061fe6 · a061fe6
1 parent 7c32b68
commit a061fe6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vllm/utils.py b/vllm/utils.py
@@ -1540,9 +1540,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]