def _save_results()

in src/qbsolv_community.py [0:0]

• 15 lines of code
• 5 McCabe index (conditional complexity)

    def _save_results(track_dict, results_graph, output_parent, result_file):
        
        results_df = pd.DataFrame.from_dict(results_graph)
        for top_key, item_list in track_dict.items():
            output_subdir = f'{output_parent}/{top_key}'
            if not os.path.exists(output_subdir):
                os.makedirs(output_subdir)
            for item in item_list:
                out_loc = f'{output_subdir}/{item[0]}'
                print(f'Writing to disk: {out_loc}')
                if top_key == 'graphs':
                    nx.write_gpickle(item[1], out_loc)
                else:
                    pickle.dump(item[1], open(out_loc, 'wb'))

        print(f'Writing to disk: {output_parent}/{result_file}')
        results_df.to_csv(f'{output_parent}/{result_file}', index=False)