From 52b7d8d391388ca11ed2995b51aecceed5b39087 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 9 Apr 2024 23:01:53 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- nemo/export/trt_llm/nemo_utils.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/nemo/export/trt_llm/nemo_utils.py b/nemo/export/trt_llm/nemo_utils.py index da53890d844e..e3c226ef639e 100644 --- a/nemo/export/trt_llm/nemo_utils.py +++ b/nemo/export/trt_llm/nemo_utils.py @@ -197,9 +197,7 @@ def nemo_llm_to_model_config( def to_word_list_format( - word_dict: List[List[str]], - tokenizer=None, - ref_str="Green tea icecream", + word_dict: List[List[str]], tokenizer=None, ref_str="Green tea icecream", ): ''' format of word_dict