Skip to content

Commit

Permalink
Update src/c++/perf_analyzer/genai-perf/tests/test_llm_metrics.py
Browse files Browse the repository at this point in the history
Co-authored-by: Hyunjae Woo <[email protected]>
  • Loading branch information
tgerdesnv and nv-hwoo committed Mar 22, 2024
1 parent 46a9041 commit 2cb6062
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/c++/perf_analyzer/genai-perf/tests/test_llm_metrics.py
Original file line number Diff line number Diff line change
Expand Up @@ -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")
Expand Down

0 comments on commit 2cb6062

Please sign in to comment.