Skip to content

Commit 76985bd

Browse files
authored
fix: small typing issue (#250)
1 parent e1bf00d commit 76985bd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

model2vec/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -331,7 +331,7 @@ def encode_as_sequence(
331331
return out_array[0]
332332
return out_array
333333

334-
def _encode_batch_as_sequence(self, sentences: list[str], max_length: int | None) -> list[np.ndarray]:
334+
def _encode_batch_as_sequence(self, sentences: Sequence[str], max_length: int | None) -> list[np.ndarray]:
335335
"""Encode a batch of sentences as a sequence."""
336336
ids = self.tokenize(sentences=sentences, max_length=max_length)
337337
out: list[np.ndarray] = []

0 commit comments

Comments
 (0)