Skip to content

Bug: validation of DocList inside a Doc raises error #1474

@AnneYang720

Description

@AnneYang720

code example:

import numpy as np

from docarray import BaseDoc, DocList
from docarray.typing import NdArray

class SimpleDoc(BaseDoc):
    simple_tens: NdArray[10]

class NestedDoc(BaseDoc):
    docs: DocList[SimpleDoc]

nested_docs = NestedDoc(
    docs=DocList[SimpleDoc]([SimpleDoc(simple_tens=np.ones(10)) for j in range(2)]),
)

a = nested_docs
assert a == nested_docs # raises ValueError

The SimpleDoc inside is turned to a dict by self.dict() in __eq__ of BaseDoc.
But then _validate_docs of class DocList will raise ValueError {...} is not a <class SimpleDoc>.

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type
No fields configured for issues without a type.

Projects

Status
Done

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions