auto_extract.py

# -*- coding: utf-8 -*-
import os
import argparse

import pcode.utils.op_files as op_files
from pcode.tools.show_results import load_raw_info_from_experiments

"""parse and define arguments for different tasks."""


def get_args():
    # feed them to the parser.
    parser = argparse.ArgumentParser(description="Extract results.")

    # add arguments.
    parser.add_argument("--in_dir", type=str)
    parser.add_argument("--out_name", type=str, default="summary.pickle")

    # parse aˇˇrgs.
    args = parser.parse_args()

    # an argument safety check.
    check_args(args)
    return args


def check_args(args):
    assert args.in_dir is not None

    # define out path.
    args.out_path = os.path.join(args.in_dir, args.out_name)


"""write the results to path."""


def main(args):
    # save the parsed results to path.
    op_files.write_pickle(load_raw_info_from_experiments(args.in_dir), args.out_path)


if __name__ == "__main__":
    args = get_args()

    main(args)