refactor: use safe_issubclass (docarray#1778)

Joan Fontanals · web-flow · commit 08ca686dd397 · 2023-09-06T19:35:08.000+02:00
Signed-off-by: Joan Fontanals Martinez &lt;joan.martinez@jina.ai&gt;
diff --git a/docarray/array/doc_vec/doc_vec.py b/docarray/array/doc_vec/doc_vec.py
@@ -214,7 +214,7 @@ def _check_doc_field_not_none(field_name, doc):
 
                         stacked: tf.Tensor = tf.stack(tf_stack)
                         tensor_columns[field_name] = TensorFlowTensor(stacked)
-                elif jnp_available and issubclass(field_type, JaxArray):
+                elif jnp_available and safe_issubclass(field_type, JaxArray):
                     if first_doc_is_none:
                         _verify_optional_field_of_docs(docs)
                         tensor_columns[field_name] = None
diff --git a/docarray/typing/tensor/abstract_tensor.py b/docarray/typing/tensor/abstract_tensor.py
@@ -95,7 +95,7 @@ def __instancecheck__(cls, instance):
                     for candidate in type(instance).__mro__
                 )
             return any(
-                issubclass(candidate, cls) for candidate in type(instance).__mro__
+                safe_issubclass(candidate, cls) for candidate in type(instance).__mro__
             )
         return super().__instancecheck__(instance)
 

Original file line number	Diff line number	Diff line change
`@@ -95,7 +95,7 @@ def __instancecheck__(cls, instance):`
`95`	`95`	`for candidate in type(instance).__mro__`
`96`	`96`	`)`
`97`	`97`	`return any(`
`98`		`- issubclass(candidate, cls) for candidate in type(instance).__mro__`
	`98`	`+ safe_issubclass(candidate, cls) for candidate in type(instance).__mro__`
`99`	`99`	`)`
`100`	`100`	`return super().__instancecheck__(instance)`
`101`	`101`