From 2c7e9d5fe07fb9030ea967d4b148f4f66f07f7d7 Mon Sep 17 00:00:00 2001 From: stephantul Date: Tue, 27 May 2025 10:42:07 +0200 Subject: [PATCH] fix: small typing issue --- model2vec/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model2vec/model.py b/model2vec/model.py index 200b95d..2ebcc14 100644 --- a/model2vec/model.py +++ b/model2vec/model.py @@ -331,7 +331,7 @@ def encode_as_sequence( return out_array[0] return out_array - def _encode_batch_as_sequence(self, sentences: list[str], max_length: int | None) -> list[np.ndarray]: + def _encode_batch_as_sequence(self, sentences: Sequence[str], max_length: int | None) -> list[np.ndarray]: """Encode a batch of sentences as a sequence.""" ids = self.tokenize(sentences=sentences, max_length=max_length) out: list[np.ndarray] = []