replace usage of issubclass with safe_issubclass in `BaseDocWitho… (

docarray#1904) Signed-off-by: YuXuan Tay <wyextay@gmail.com> Co-authored-by: Joan Fontanals <joan.martinez@jina.ai>
Jibola · Aug 17, 2024 · e6802a2 · e6802a2
1 parent 8f4ba7c
commit e6802a2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
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:]),