diff --git a/docarray/documents/text.py b/docarray/documents/text.py index 0e3edbbcec2..98a335fb268 100644 --- a/docarray/documents/text.py +++ b/docarray/documents/text.py @@ -43,6 +43,7 @@ class TextDoc(BaseDocument): from docarray.typing import AnyEmbedding from typing import Optional + # extend it class MyText(Text): second_embedding: Optional[AnyEmbedding] @@ -62,6 +63,7 @@ class MyText(Text): from docarray import BaseDocument from docarray.documents import ImageDoc, TextDoc + # compose it class MultiModalDoc(BaseDocument): image_doc: Image @@ -96,10 +98,10 @@ class MultiModalDoc(BaseDocument): doc == doc2 # False, their ids are not equivalent """ - text: Optional[str] = None - url: Optional[TextUrl] = None - embedding: Optional[AnyEmbedding] = None - bytes: Optional[bytes] = None + text: Optional[str] + url: Optional[TextUrl] + embedding: Optional[AnyEmbedding] + bytes: Optional[bytes] def __init__(self, text: Optional[str] = None, **kwargs): if 'text' not in kwargs: diff --git a/docarray/documents/video.py b/docarray/documents/video.py index debd93e3540..407d966fb3c 100644 --- a/docarray/documents/video.py +++ b/docarray/documents/video.py @@ -102,7 +102,7 @@ class MultiModalDoc(BaseDocument): tensor: Optional[VideoTensor] key_frame_indices: Optional[AnyTensor] embedding: Optional[AnyEmbedding] - bytes: Optional[bytes] = None + bytes: Optional[bytes] @classmethod def validate(