Blame for code/tt_moge/reference/utils3d/torch/__init__.py

import itertools
from typing import TYPE_CHECKING
from ..helpers import lazy_import_all_from

try:
    import torch
except ImportError:
    raise ImportError("Pytorch is not installed. Please install Pytorch to use the torch functions of utils3d.")

module_members = {}

for module_name in ['utils', 'transforms', 'pose', 'segment_ops', 'mesh', 'maps', 'rasterization']:
    module_members[module_name] = lazy_import_all_from(globals(), '.' + module_name)

__all__ = list(itertools.chain(*module_members.values()))


if TYPE_CHECKING:
    from .utils import *
    from .transforms import *
    from .segment_ops import *
    from .mesh import *
    from .maps import *
    from .rasterization import *
    from .segment_ops import *
    from .pose import *