From b762a745a2ec6463c3d6bedae32442c1b5ddb025 Mon Sep 17 00:00:00 2001
From: Morgan Du <morgandu@google.com>
Date: Mon, 3 Jun 2024 22:36:04 +0000
Subject: [PATCH] fix typo

---
 benchmarks/benchmark_serving.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/benchmarks/benchmark_serving.py b/benchmarks/benchmark_serving.py
index d68be2d2..07b36a84 100644
--- a/benchmarks/benchmark_serving.py
+++ b/benchmarks/benchmark_serving.py
@@ -829,7 +829,7 @@ def main(args: argparse.Namespace):
       "--warmup-mode",
       type=str,
       default="none",
-      choices=["none", "sample", "full"],
+      choices=["none", "sampled", "full"],
       help="Whether to warmup first, and set the warmup mode",
   )
   parser.add_argument(