Skip to content

Commit 4bf5b73

Browse files
remove redundant if/else block
1 parent 81ba647 commit 4bf5b73

File tree

1 file changed

+2
-8
lines changed

1 file changed

+2
-8
lines changed

tests/integration/test_vectorizers.py

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -340,10 +340,7 @@ async def aembed_many_func(texts):
340340
@pytest.mark.asyncio
341341
async def test_vectorizer_aembed(avectorizer):
342342
text = "This is a test sentence."
343-
if isinstance(avectorizer, VoyageAITextVectorizer):
344-
embedding = await avectorizer.aembed(text)
345-
else:
346-
embedding = await avectorizer.aembed(text)
343+
embedding = await avectorizer.aembed(text)
347344

348345
assert isinstance(embedding, list)
349346
assert len(embedding) == avectorizer.dims
@@ -352,10 +349,7 @@ async def test_vectorizer_aembed(avectorizer):
352349
@pytest.mark.asyncio
353350
async def test_vectorizer_aembed_many(avectorizer):
354351
texts = ["This is the first test sentence.", "This is the second test sentence."]
355-
if isinstance(avectorizer, VoyageAITextVectorizer):
356-
embeddings = await avectorizer.aembed_many(texts)
357-
else:
358-
embeddings = await avectorizer.aembed_many(texts)
352+
embeddings = await avectorizer.aembed_many(texts)
359353

360354
assert isinstance(embeddings, list)
361355
assert len(embeddings) == len(texts)

0 commit comments

Comments
 (0)