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] = []