From dc5114a6abae69fcf6bb1de34fd76b2511a1ba50 Mon Sep 17 00:00:00 2001 From: Hyunjae Woo Date: Fri, 1 Dec 2023 10:04:55 -0800 Subject: [PATCH] Change to direct string comparison --- src/c++/perf_analyzer/docs/examples/profile.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/c++/perf_analyzer/docs/examples/profile.py b/src/c++/perf_analyzer/docs/examples/profile.py index 9793adb3e..7b79c9848 100644 --- a/src/c++/perf_analyzer/docs/examples/profile.py +++ b/src/c++/perf_analyzer/docs/examples/profile.py @@ -569,7 +569,7 @@ def construct_trtllm_input_data(args): def main(args): if args.backend == "trtllm": input_data = construct_trtllm_input_data(args) - elif args.backend in "vllm": + elif args.backend == "vllm": input_data = construct_vllm_input_data(args) else: raise ValueError(