@@ -261,9 +261,9 @@ def bad_return_type(text: str) -> str:
261261)
262262def 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_):
291291def 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