def parser_for_pipeline(pipeline):
    if pipeline == 'train':
        return ModelParser(
            'estimator',
            'train',