diff --git a/docarray/array/__init__.py b/docarray/array/__init__.py index bd3461f4826..e670b680fe5 100644 --- a/docarray/array/__init__.py +++ b/docarray/array/__init__.py @@ -1,3 +1,3 @@ -from docarray.array.documentarray import DocumentArray +from docarray.array.array import DocumentArray __all__ = ['DocumentArray'] diff --git a/docarray/array/documentarray.py b/docarray/array/array.py similarity index 94% rename from docarray/array/documentarray.py rename to docarray/array/array.py index 7963535c006..d15f04a7c11 100644 --- a/docarray/array/documentarray.py +++ b/docarray/array/array.py @@ -14,7 +14,7 @@ class DocumentArray( BaseNode, ): """ - a _GenericDocumentArray is a list-like container of Document of the same schema + a DocumentArray is a list-like container of Document of the same schema :param docs: iterable of Document """ diff --git a/tests/units/array/test_mixins/test_attribute.py b/tests/units/array/test_mixins/test_attribute.py index 5eabfb8b7ef..cf955702a09 100644 --- a/tests/units/array/test_mixins/test_attribute.py +++ b/tests/units/array/test_mixins/test_attribute.py @@ -5,7 +5,7 @@ from docarray.typing import Tensor -def test_get_bulk_attributes(): +def test_get_bulk_attributes_function(): class Mmdoc(BaseDocument): text: str tensor: Tensor