From 2cb60620000903c29c37a48a207a6b0c302b1d37 Mon Sep 17 00:00:00 2001 From: Timothy Gerdes <50968584+tgerdesnv@users.noreply.github.com> Date: Fri, 22 Mar 2024 10:43:20 -0500 Subject: [PATCH] Update src/c++/perf_analyzer/genai-perf/tests/test_llm_metrics.py Co-authored-by: Hyunjae Woo <107147848+nv-hwoo@users.noreply.github.com> --- src/c++/perf_analyzer/genai-perf/tests/test_llm_metrics.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/c++/perf_analyzer/genai-perf/tests/test_llm_metrics.py b/src/c++/perf_analyzer/genai-perf/tests/test_llm_metrics.py index cd2af3e40..57982fd00 100755 --- a/src/c++/perf_analyzer/genai-perf/tests/test_llm_metrics.py +++ b/src/c++/perf_analyzer/genai-perf/tests/test_llm_metrics.py @@ -84,7 +84,8 @@ def write(content): def test_csv_output(self, mock_read_write) -> None: """ - Collect LLM metrics from profile export data and confirm correct values are printed in csv + Collect LLM metrics from profile export data and confirm correct values are + printed in csv. """ tokenizer = AutoTokenizer.from_pretrained("gpt2")