text2motion/datasets/__init__.py

from .dataloader import build_dataloader
from .dataset import Text2MotionDataset
from .evaluator import (EvaluationDataset, EvaluatorModelWrapper,
                        get_dataset_motion_loader, get_motion_loader)

from .utils import drop_shapes_from_motion_arr
# from .rendering import render_meshes
from .motionx_explorer import (load_data_as_dict,
                               motion_arr_to_dict, smplx_dict_to_array, to_smplx_dict)

__all__ = [
    'Text2MotionDataset', 'EvaluationDataset', 'build_dataloader',
    'get_dataset_motion_loader', 'get_motion_loader', 'EvaluatorModelWrapper',
    'load_data_as_dict', 'motion_arr_to_dict', 'smplx_dict_to_array',
    'drop_shapes_from_motion_arr', 'render_meshes', 'to_smplx_dict'
]