GeezerVison is showing github.com — docarray/docarray/store/__init__.py at main · docarray/docarray

This repository was archived by the owner on Sep 9, 2026. It is now read-only.

Expand file tree

init.py

Copy path

More file actions

33 lines (25 loc) · 851 Bytes

Copy path

33 lines (25 loc) · 851 Bytes

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

import types

from typing import TYPE_CHECKING

from docarray.store.file import FileDocStore

from docarray.utils._internal.misc import (

_get_path_from_docarray_root_level,

import_library,

)

if TYPE_CHECKING:

from docarray.store.s3 import S3DocStore # noqa: F401

__all__ = ['FileDocStore']

def __getattr__(name: str):

lib: types.ModuleType

if name == 'S3DocStore':

import_library('smart_open', raise_error=True)

import_library('botocore', raise_error=True)

import_library('boto3', raise_error=True)

import docarray.store.s3 as lib

else:

raise ImportError(

f'cannot import name \'{name}\' from \'{_get_path_from_docarray_root_level(__file__)}\''

)

store_cls = getattr(lib, name)

if name not in __all__:

__all__.append(name)

return store_cls