from __future__ import annotations

from .analysis.tasks import build_analysis_task
from .dataset.tasks import build_dataset_task
from .mmseg.tasks import build_mmseg_task
from .segmodel.tasks import build_segmodel_task
from .system.tasks import build_system_task
from .yolo.tasks import build_yolo_task
from .visual.tasks import build_visual_task


def build_module_task(job_type: str, params: dict, conda_env: str):
    for builder in (
        build_dataset_task,
        build_segmodel_task,
        build_yolo_task,
        build_visual_task,
        build_mmseg_task,
        build_analysis_task,
        build_system_task,
    ):
        spec = builder(job_type, params, conda_env)
        if spec is not None:
            return spec
    return None