def main()

in Utils/get_tokens_and_time.py [0:0]

• 16 lines of code
• 7 McCabe index (conditional complexity)

def main(models=None, langs=None, experiments=default_experiments):
    header = "Experiment,Category,Language,Models,Dataset,Complexity,Size,Attempt,Input,Reasons,Output,Time,Accuracy,Completeness\n"
    for model in models:
        for lang in langs:
            data = header
            for experiment in experiments:
                current_path = results_path / 'Output' / model / lang / experiment
                if not os.path.isdir(current_path): continue
                for experiment_folder in os.listdir(current_path):
                    if experiment_folder.startswith('.'): continue
                    experiment_folder_path = current_path / experiment_folder
                    data += process_directory(experiment_folder_path, model, experiment)

            output_path = results_path / 'Output' / model / lang / 'summary.csv'
            with open(output_path, 'w') as f:
                f.write(data)
            print(f'Summary written successfully for {model} and lang {lang} to {output_path}')