Change to direct string comparison

triton-inference-server · Dec 1, 2023 · dc5114a · dc5114a
1 parent 7205793
commit dc5114a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(