File tree Expand file tree Collapse file tree 1 file changed +2
-8
lines changed
Expand file tree Collapse file tree 1 file changed +2
-8
lines changed Original file line number Diff line number Diff line change @@ -340,10 +340,7 @@ async def aembed_many_func(texts):
340340@pytest .mark .asyncio
341341async 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
353350async 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 )
You can’t perform that action at this time.
0 commit comments