def main()

in prepare_results.py [0:0]

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

def main(args):
    all_csvs = sorted(glob.glob(f"{args.base_path}/*.csv"))
    all_csvs = [os.path.join(args.base_path, x) for x in all_csvs]
    is_pixart = "PixArt-alpha" in all_csvs[0]
    collate_csv(all_csvs, args.final_csv_filename, is_pixart=is_pixart)

    if args.push_to_hub:
        upload_file(
            repo_id=REPO_ID,
            path_in_repo=args.final_csv_filename,
            path_or_fileobj=args.final_csv_filename,
            repo_type="dataset",
        )
        print(
            f"CSV successfully uploaded. Find it here: https://huggingface.co/datasets/{REPO_ID}/blob/main/{args.final_csv_filename}"
        )

    if args.plot_title is not None:
        df = pd.read_csv(args.final_csv_filename)
        prepare_plot(df, args)