Function write_data

experiments/human_eval/draw_model_eval_ggbench_style.py:223–235

(rows: list[dict[str, object]])

def write_data(rows: list[dict[str, object]]) -> None:
    with CSV_OUT.open("w", encoding="utf-8", newline="") as fh:
        writer = csv.DictWriter(fh, fieldnames=["model", "group", "ModelAutoScore", "ModelHumanScore"])
        writer.writeheader()
        for row in rows:
            writer.writerow(
                {
                    "model": row["model"],
                    "group": row["group"],
                    "ModelAutoScore": f"{float(row['auto']):.2f}",
                    "ModelHumanScore": f"{float(row['human']):.2f}",
                }
            )


def main() -> None: