Skip to content

Commit 01f0876

Browse files
nv-hwoomc-nv
authored andcommitted
Change to _ instead of removing
1 parent f7745e4 commit 01f0876

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

src/c++/perf_analyzer/genai-perf/genai_perf/parser.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,7 @@ def _set_artifact_paths(args: argparse.Namespace) -> argparse.Namespace:
141141
if args.artifact_dir == Path(DEFAULT_ARTIFACT_DIR):
142142
# Preprocess Huggingface model names that include '/' in their model name.
143143
if (args.model is not None) and ("/" in args.model):
144-
filtered_name = args.model.split("/")[-1]
144+
filtered_name = "_".join(args.model.split("/"))
145145
logger.info(
146146
f"Model name '{args.model}' cannot be used to create artifact "
147147
f"directory. Instead, '{filtered_name}' will be used."

src/c++/perf_analyzer/genai-perf/tests/test_cli.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -251,19 +251,19 @@ def test_default_profile_export_filepath(
251251
"arg, expected_path",
252252
[
253253
(
254-
["--model", "company/test_model"],
255-
"artifacts/test_model-triton-tensorrtllm-concurrency1",
254+
["--model", "strange/test_model"],
255+
"artifacts/strange_test_model-triton-tensorrtllm-concurrency1",
256256
),
257257
(
258258
[
259259
"--model",
260-
"company/test_model",
260+
"hello/world/test_model",
261261
"--service-kind",
262262
"openai",
263263
"--endpoint-type",
264264
"chat",
265265
],
266-
"artifacts/test_model-openai-chat-concurrency1",
266+
"artifacts/hello_world_test_model-openai-chat-concurrency1",
267267
),
268268
],
269269
)

0 commit comments

Comments
 (0)