Skip to content

Commit c5e253b

Browse files
uncomment vectorizer subclasses
1 parent a11ceda commit c5e253b

File tree

1 file changed

+16
-16
lines changed

1 file changed

+16
-16
lines changed

tests/integration/test_vectorizers.py

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)