attacks/AIM/src/gat/datasets/__init__.py

from .builder import build_dataset, list_datasets
from .cub import cub
from .imagenet import imagenet

__all__ = ['build_dataset', 'list_datasets', 'imagenet', 'cub']