Try passing in columns explicitly when writing csv

JasonGross · Oct 31, 2024 · d1673af · d1673af
1 parent c914719
commit d1673af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/notebooks_jason/max_of_K_all_models.py b/notebooks_jason/max_of_K_all_models.py
@@ -782,7 +782,7 @@ def update_csv_with_rows(
             subset=subset, keep="last"
         )
     results = results.sort_values(subset)
-    results.to_csv(csv_path, index=False)
+    results.to_csv(csv_path, index=False, columns=columns)
     if save_to_hf:
         results.to_parquet(csv_to_hf_path(csv_path, ext="parquet"))
         results.to_csv(csv_to_hf_path(csv_path, ext="csv"), index=False)