flowae/datasets/__init__.py

from .datasets import register, make
from . import image_folder, class_folder, webdataset, class_folder_audio
from . import wrapper_cae, wrapper_audio_cae