import argparse

import json

from pipelines import pipelines

def parse_args():

    parser = argparse.ArgumentParser()

    parser.add_argument("--config", required=True)

    return parser.parse_args()

if __name__ == "__main__":

    args = parse_args()

    config = json.loads(open(args.config).read())

    pipeline_type = getattr(pipelines, config["type"])

    print("Trainer: ", config["type"], pipeline_type)

    pipeline = pipeline_type(config)

    pipeline.run_pipeline()