🐛 fix validation error formatting

Signed-off-by: Joe Runde <[email protected]>
IBM · Mar 27, 2024 · f477ba3 · f477ba3
1 parent c9d76f8
commit f477ba3
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/tests/tgis/__init__.py b/tests/tgis/__init__.py
diff --git a/vllm/entrypoints/grpc/validation.py b/vllm/entrypoints/grpc/validation.py
@@ -39,7 +39,7 @@ class TGISValidationError(str, Enum):
 
     def error(self, *args, **kwargs):
         """Raises a ValueError with a nicely formatted string"""
-        raise ValueError(self.format(*args, **kwargs))
+        raise ValueError(self.value.format(*args, **kwargs))
 
 
 def validate_input(sampling_params: SamplingParams, token_num: int, max_model_len: int):