def task_type(task):
    valid_tasks = ['RUN_COMMAND', 'AUTOMATION', 'LAMBDA', 'STEP_FUNCTION']
    if (task not in valid_tasks):
        raise ValueError(('TaskType must be one of: "%s"' % ', '.join(valid_tasks)))
    return task
def task_type(task):
    valid_tasks = ['RUN_COMMAND', 'AUTOMATION', 'LAMBDA', 'STEP_FUNCTION']
    if (task not in valid_tasks):
        raise ValueError(('TaskType must be one of: "%s"' % ', '.join(valid_tasks)))
    return task