from argparse import ArgumentParser

def getArgs():
    parser = ArgumentParser()
    parser.add_argument("type")
    parser.add_argument("datasetPath")
    parser.add_argument("--weightsFile")
    args = parser.parse_args()
    return args

def main():
    args = getArgs()

    if args.type == "train":
        pass
    elif args.type == "test":
        pass
    elif args.type == "export_csv":
        pass
    else
        assert False

if __name__ == "__main__":
    main()