replace usage of issubclass with safe_issubclass in `BaseDocWitho… (#1904)

yxtay · Joan Fontanals · web-flow · commit e6802a2b8fcc · 2024-08-17T09:09:23.000+02:00
Signed-off-by: YuXuan Tay &lt;wyextay@gmail.com&gt;
Co-authored-by: Joan Fontanals &lt;joan.martinez@jina.ai&gt;
diff --git a/docarray/base_doc/doc.py b/docarray/base_doc/doc.py
@@ -326,7 +326,7 @@ def _exclude_doclist(
             from docarray.array.any_array import AnyDocArray
 
             type_ = self._get_field_annotation(field)
-            if isinstance(type_, type) and issubclass(type_, AnyDocArray):
+            if isinstance(type_, type) and safe_issubclass(type_, AnyDocArray):
                 doclist_exclude_fields.append(field)
 
         original_exclude = exclude
diff --git a/docarray/index/backends/milvus.py b/docarray/index/backends/milvus.py
@@ -192,7 +192,7 @@ def python_type_to_db_type(self, python_type: Type) -> Any:
             AbstractTensor: DataType.FLOAT_VECTOR,
         }
 
-        if issubclass(python_type, ID):
+        if safe_issubclass(python_type, ID):
             return DataType.VARCHAR
 
         for py_type, db_type in type_map.items():
@@ -665,7 +665,7 @@ def find_batched(
         if search_field:
             if '__' in search_field:
                 fields = search_field.split('__')
-                if issubclass(self._schema._get_field_annotation(fields[0]), AnyDocArray):  # type: ignore
+                if safe_issubclass(self._schema._get_field_annotation(fields[0]), AnyDocArray):  # type: ignore
                     return self._subindices[fields[0]].find_batched(
                         queries,
                         search_field='__'.join(fields[1:]),
