diff --git a/docarray/__init__.py b/docarray/__init__.py index 903c0947d43..ce2152c8315 100644 --- a/docarray/__init__.py +++ b/docarray/__init__.py @@ -4,6 +4,7 @@ from docarray.array import DocList, DocVec from docarray.base_doc.doc import BaseDoc +from docarray.utils._internal.misc import _get_path_from_docarray_root_level __all__ = ['BaseDoc', 'DocList', 'DocVec'] @@ -13,3 +14,17 @@ formatter = logging.Formatter("%(levelname)s - %(name)s - %(message)s") handler.setFormatter(formatter) logger.addHandler(handler) + + +def __getattr__(name: str): + if name in ['Document', 'DocumentArray']: + raise ImportError( + f'Cannot import name \'{name}\' from \'{_get_path_from_docarray_root_level(__file__)}\'.\n' + f'The object named \'{name}\' does not exist anymore in this version of docarray.\n' + f'If you still want to use \'{name}\' please downgrade to version <=0.21.0 ' + f'with: `pip install -U docarray==0.21.0`.' + ) + else: + raise ImportError( + f'cannot import name \'{name}\' from \'{_get_path_from_docarray_root_level(__file__)}\'' + )