From 5fd8f4457442b17a20a709ff8967bec4a1ed50c3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fl=C3=A1via=20B=C3=A9o?= <119421251+flaviabeo@users.noreply.github.com> Date: Wed, 24 Jul 2024 15:14:01 -0400 Subject: [PATCH] Update caikit_nlp/modules/text_embedding/embedding.py MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: Evaline Ju <69598118+evaline-ju@users.noreply.github.com> Signed-off-by: Flávia Béo <119421251+flaviabeo@users.noreply.github.com> --- caikit_nlp/modules/text_embedding/embedding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/caikit_nlp/modules/text_embedding/embedding.py b/caikit_nlp/modules/text_embedding/embedding.py index 5db48895..3aaf74f5 100644 --- a/caikit_nlp/modules/text_embedding/embedding.py +++ b/caikit_nlp/modules/text_embedding/embedding.py @@ -213,7 +213,7 @@ def run_tokenizer( """Run tokenization task against the model Args: - text: str + text: str Text to tokenize Returns: TokenizationResults