Skip to content

Commit e5d3d63

Browse files
[Benchmark] Fix terminal colors in benchmark_serving_multi_turn (python 3.12) (vllm-project#22730)
Signed-off-by: daniels <[email protected]>
1 parent 3d9d40e commit e5d3d63

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

benchmarks/multi_turn/bench_utils.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
from enum import Enum
55

66

7-
class Color(str, Enum):
7+
class Color(Enum):
88
RED = "\033[91m"
99
GREEN = "\033[92m"
1010
BLUE = "\033[94m"
@@ -13,6 +13,9 @@ class Color(str, Enum):
1313
YELLOW = "\033[93m"
1414
RESET = "\033[0m"
1515

16+
def __str__(self):
17+
return self.value
18+
1619

1720
TEXT_SEPARATOR = "-" * 100
1821

0 commit comments

Comments
 (0)