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(