from utils.config import Config

from .train_SP_pipeline import Train_SP_Pipeline
from .test_SP_pipeline_base import Test_SP_Pipeline_base


def get_pipeline(config: Config):
    pipelines = {
        'SP_train': Train_SP_Pipeline,
        'SP_test_base': Test_SP_Pipeline_base,
    }

    return pipelines[config.pipeline.name](config)
