@@ -41,11 +41,11 @@ def embeddings_cache(client):
4141 params = [
4242 HFTextVectorizer ,
4343 OpenAITextVectorizer ,
44- # VertexAITextVectorizer,
44+ VertexAITextVectorizer ,
4545 CohereTextVectorizer ,
46- # AzureOpenAITextVectorizer,
47- # BedrockTextVectorizer,
48- # MistralAITextVectorizer,
46+ AzureOpenAITextVectorizer ,
47+ BedrockTextVectorizer ,
48+ MistralAITextVectorizer ,
4949 CustomTextVectorizer ,
5050 VoyageAITextVectorizer ,
5151 ]
@@ -418,14 +418,14 @@ def bad_return_type(text: str) -> str:
418418@pytest .mark .parametrize (
419419 "vectorizer_" ,
420420 [
421- # AzureOpenAITextVectorizer,
422- # BedrockTextVectorizer,
421+ AzureOpenAITextVectorizer ,
422+ BedrockTextVectorizer ,
423423 CohereTextVectorizer ,
424424 CustomTextVectorizer ,
425425 HFTextVectorizer ,
426- # MistralAITextVectorizer,
426+ MistralAITextVectorizer ,
427427 OpenAITextVectorizer ,
428- # VertexAITextVectorizer,
428+ VertexAITextVectorizer ,
429429 VoyageAITextVectorizer ,
430430 ],
431431)
@@ -447,14 +447,14 @@ def test_default_dtype(vectorizer_):
447447@pytest .mark .parametrize (
448448 "vectorizer_" ,
449449 [
450- # AzureOpenAITextVectorizer,
451- # BedrockTextVectorizer,
450+ AzureOpenAITextVectorizer ,
451+ BedrockTextVectorizer ,
452452 CohereTextVectorizer ,
453453 CustomTextVectorizer ,
454454 HFTextVectorizer ,
455- # MistralAITextVectorizer,
455+ MistralAITextVectorizer ,
456456 OpenAITextVectorizer ,
457- # VertexAITextVectorizer,
457+ VertexAITextVectorizer ,
458458 VoyageAITextVectorizer ,
459459 ],
460460)
@@ -480,13 +480,13 @@ def test_vectorizer_dtype_assignment(vectorizer_):
480480@pytest .mark .parametrize (
481481 "vectorizer_" ,
482482 [
483- # AzureOpenAITextVectorizer,
484- # BedrockTextVectorizer,
483+ AzureOpenAITextVectorizer ,
484+ BedrockTextVectorizer ,
485485 CohereTextVectorizer ,
486486 HFTextVectorizer ,
487- # MistralAITextVectorizer,
487+ MistralAITextVectorizer ,
488488 OpenAITextVectorizer ,
489- # VertexAITextVectorizer,
489+ VertexAITextVectorizer ,
490490 VoyageAITextVectorizer ,
491491 ],
492492)
0 commit comments