from .task import Task  # noqa
from .task_config import TaskConfig  # noqa
from .dummy_task_config import DummyTaskConfig, DummyTask  # noqa
from .distance_task_config import DistanceTaskConfig, DistanceTask  # noqa
from .one_hot_task_config import OneHotTaskConfig, OneHotTask  # noqa
from .pretrained_task_config import PretrainedTaskConfig, PretrainedTask  # noqa
from .affinities_task_config import AffinitiesTaskConfig, AffinitiesTask  # noqa
from .inner_distance_task_config import (
    InnerDistanceTaskConfig,
    InnerDistanceTask,
)  # noqa
from .hot_distance_task_config import HotDistanceTaskConfig, HotDistanceTask  # noqa
