ImageBind/imagebind/__init__.py

from imagebind import data
from imagebind.models import imagebind_model
from imagebind.models.imagebind_model import ModalityType

try:
    from imagebind.legrad_wrapper import ImageBindLeWrapper
except ImportError:  # pragma: no cover — optional LeGrad dependency
    ImageBindLeWrapper = None  # type: ignore