def get_pipeline()

in docker_images/sklearn/app/main.py [0:0]

• 8 lines of code
• 4 McCabe index (conditional complexity)

def get_pipeline(task=None, model_id=None) -> Pipeline:
    task = task or os.environ["TASK"]
    model_id = model_id or os.environ["MODEL_ID"]
    if task not in ALLOWED_TASKS:
        raise EnvironmentError(
            f"{task} is not a valid pipeline for model : {model_id} ({','.join(ALLOWED_TASKS.keys())})"
        )
    return ALLOWED_TASKS[task](model_id)