def process_results()

in scripts/plot_results.py [0:0]

• 27 lines of code
• 4 McCabe index (conditional complexity)

def process_results(dirname, commit, save_dirname): 
    subdirs = [os.path.join(dirname, subdir, commit + '/') for subdir in os.listdir(dirname) if os.path.isdir(os.path.join(dirname, subdir))]
    for subdir in subdirs:
        print(subdir)
        table, table_avg, table_hparams, table_val, table_val_avg, df = collect_results.build_table(subdir)

        # plot table_val
        plot_table(
                table=table_val, 
                dirname=subdir, 
                file_name='_'.join(subdir.split('/')[-3:-1]),
                save=True, block=False)
        # save table_val
        save_dirname_single = save_dirname + "single/"
        os.makedirs(save_dirname_single, exist_ok=True)
        results_filename = os.path.join(save_dirname_single, 'single_' + '_'.join(subdir.split('/')[-4:-1]) + ".jsonl")
        results_file = open(results_filename, "w")
        results_file.write(json.dumps(table_val))
        results_file.close()

        # plot table_val_avg
        plot_table_avg(
                table=table_val_avg, 
                dirname=subdir, 
                file_name='avg_' + '_'.join(subdir.split('/')[-3:-1]),
                save=True, block=False)
        # save table_val_avg
        save_dirname_avg = save_dirname + "avg/"
        os.makedirs(save_dirname_avg, exist_ok=True)
        results_filename = os.path.join(save_dirname_avg, 'avg_' + '_'.join(subdir.split('/')[-4:-1]) + ".jsonl")
        results_file = open(results_filename, "w")
        results_file.write(json.dumps(table_val_avg))
        results_file.close()