Skip to content

Commit e51ae91

Browse files
authored
Add an = to <= check in embedding_utils (openai#118)
1 parent 3edecbe commit e51ae91

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

openai/embeddings_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ def get_embedding(text: str, engine="text-similarity-davinci-001") -> List[float
2727
def get_embeddings(
2828
list_of_text: List[str], engine="text-similarity-babbage-001"
2929
) -> List[List[float]]:
30-
assert len(list_of_text) < 2048, "The batch size should not be larger than 2048."
30+
assert len(list_of_text) <= 2048, "The batch size should not be larger than 2048."
3131

3232
# replace newlines, which can negatively affect performance.
3333
list_of_text = [text.replace("\n", " ") for text in list_of_text]

0 commit comments

Comments
 (0)