25 lines
688 B
Python
25 lines
688 B
Python
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
|
|
|
|
|
|
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_mmseg_task,
|
|
build_analysis_task,
|
|
build_system_task,
|
|
):
|
|
spec = builder(job_type, params, conda_env)
|
|
if spec is not None:
|
|
return spec
|
|
return None
|
|
|