@@ -24,14 +24,14 @@ def skip_vectorizer() -> bool:
2424@pytest .fixture (
2525 params = [
2626 HFTextVectorizer ,
27- # OpenAITextVectorizer,
28- # VertexAITextVectorizer,
29- # CohereTextVectorizer,
30- # AzureOpenAITextVectorizer,
31- # BedrockTextVectorizer,
32- # MistralAITextVectorizer,
27+ OpenAITextVectorizer ,
28+ VertexAITextVectorizer ,
29+ CohereTextVectorizer ,
30+ AzureOpenAITextVectorizer ,
31+ BedrockTextVectorizer ,
32+ MistralAITextVectorizer ,
3333 CustomTextVectorizer ,
34- # VoyageAITextVectorizer,
34+ VoyageAITextVectorizer ,
3535 ]
3636)
3737def vectorizer (request , skip_vectorizer ):
@@ -253,20 +253,20 @@ def bad_return_type(text: str) -> str:
253253@pytest .mark .parametrize (
254254 "vector_class" ,
255255 [
256- # AzureOpenAITextVectorizer,
257- # BedrockTextVectorizer,
258- # CohereTextVectorizer,
256+ AzureOpenAITextVectorizer ,
257+ BedrockTextVectorizer ,
258+ CohereTextVectorizer ,
259259 CustomTextVectorizer ,
260260 HFTextVectorizer ,
261- # MistralAITextVectorizer,
262- # OpenAITextVectorizer,
263- # VertexAITextVectorizer,
264- # VoyageAITextVectorizer,
261+ MistralAITextVectorizer ,
262+ OpenAITextVectorizer ,
263+ VertexAITextVectorizer ,
264+ VoyageAITextVectorizer ,
265265 ],
266266)
267267def test_dtypes (vector_class , skip_vectorizer ):
268- # if skip_vectorizer:
269- # pytest.skip("Skipping vectorizer instantiation...")
268+ if skip_vectorizer :
269+ pytest .skip ("Skipping vectorizer instantiation..." )
270270
271271 # test dtype defaults to float32
272272 if issubclass (vector_class , CustomTextVectorizer ):
@@ -312,11 +312,11 @@ def test_dtypes(vector_class, skip_vectorizer):
312312
313313@pytest .fixture (
314314 params = [
315- # OpenAITextVectorizer,
316- # BedrockTextVectorizer,
317- # MistralAITextVectorizer,
315+ OpenAITextVectorizer ,
316+ BedrockTextVectorizer ,
317+ MistralAITextVectorizer ,
318318 CustomTextVectorizer ,
319- # VoyageAITextVectorizer,
319+ VoyageAITextVectorizer ,
320320 ]
321321)
322322def avectorizer (request , skip_vectorizer ):
0 commit comments