Skip to content

Commit 3176bd3

Browse files
authored
[None][fix] Fix UnboundLocalError. (#8756)
Signed-off-by: Yuxian Qiu <[email protected]>
1 parent ae57738 commit 3176bd3

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

tests/integration/defs/perf/test_perf.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1646,7 +1646,6 @@ def get_trtllm_bench_command(self, engine_dir):
16461646
benchmark_cmd += [f"--streaming"]
16471647
#use default yaml config
16481648
if self._config.backend == "pytorch":
1649-
import yaml
16501649
pytorch_config_path = os.path.join(engine_dir,
16511650
"extra-llm-api-config.yml")
16521651
if not os.path.exists(pytorch_config_path):
@@ -1658,7 +1657,6 @@ def get_trtllm_bench_command(self, engine_dir):
16581657
yaml.dump(config, f, default_flow_style=False)
16591658
benchmark_cmd += [f"--extra_llm_api_options={pytorch_config_path}"]
16601659
elif self._config.backend == "_autodeploy":
1661-
import yaml
16621660
autodeploy_config_path = os.path.join(engine_dir,
16631661
"extra_llm_api_options.yaml")
16641662
if not os.path.exists(autodeploy_config_path):

0 commit comments

Comments
 (0)