diff --git a/libs/text-splitters/langchain_text_splitters/sentence_transformers.py b/libs/text-splitters/langchain_text_splitters/sentence_transformers.py index 6934f063d61..c52b78a6460 100644 --- a/libs/text-splitters/langchain_text_splitters/sentence_transformers.py +++ b/libs/text-splitters/langchain_text_splitters/sentence_transformers.py @@ -19,7 +19,8 @@ class SentenceTransformersTokenTextSplitter(TextSplitter): super().__init__(**kwargs, chunk_overlap=chunk_overlap) try: - from transformers import AutoConfig, AutoTokenizer + from transformers import AutoConfig # type: ignore[import] + from transformers import AutoTokenizer # type: ignore[import] except ImportError: msg = ( "Could not import transformers python package. " @@ -110,3 +111,4 @@ class SentenceTransformersTokenTextSplitter(TextSplitter): truncation="do_not_truncate", ) return cast("list[int]", token_ids_with_start_and_end_token_ids) +