Skip to content

Commit 56931ac

Browse files
fix test
1 parent 7f3a0fe commit 56931ac

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/integration/test_vectorizers.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -261,9 +261,9 @@ def bad_return_type(text: str) -> str:
261261
)
262262
def test_default_dtype(vectorizer_):
263263
# test dtype defaults to float32
264-
if isinstance(vectorizer_, CustomTextVectorizer):
264+
if issubclass(vectorizer_, CustomTextVectorizer):
265265
vectorizer = vectorizer_(embed=lambda x, input_type=None: [1.0, 2.0, 3.0])
266-
elif isinstance(vectorizer_, AzureOpenAITextVectorizer):
266+
elif issubclass(vectorizer_, AzureOpenAITextVectorizer):
267267
vectorizer = vectorizer_(
268268
model=os.getenv("AZURE_OPENAI_DEPLOYMENT_NAME", "text-embedding-ada-002")
269269
)
@@ -291,9 +291,9 @@ def test_default_dtype(vectorizer_):
291291
def test_vectorizer_dtype_assignment(vectorizer_):
292292
# test initializing dtype in constructor
293293
for dtype in ["float16", "float32", "float64", "bfloat16", "int8", "uint8"]:
294-
if isinstance(vectorizer_, CustomTextVectorizer):
294+
if issubclass(vectorizer_, CustomTextVectorizer):
295295
vectorizer = vectorizer_(embed=lambda x: [1.0, 2.0, 3.0], dtype=dtype)
296-
elif isinstance(vectorizer_, AzureOpenAITextVectorizer):
296+
elif issubclass(vectorizer_, AzureOpenAITextVectorizer):
297297
vectorizer = vectorizer_(
298298
model=os.getenv(
299299
"AZURE_OPENAI_DEPLOYMENT_NAME", "text-embedding-ada-002"

0 commit comments

Comments
 (0)