| 70 | |
| 71 | |
| 72 | def save_data(dataset, data, save_path): |
| 73 | dataset["preds"] = data["preds"] |
| 74 | dataset["em"] = data["em"] |
| 75 | dataset["f1"] = data["f1"] |
| 76 | dataset["acc"] = data["acc"] |
| 77 | dataset["wall_time"] = data["wall_time"] |
| 78 | dataset["total_tokens"] = data["total_tokens"] |
| 79 | dataset["steps"] = data["steps"] |
| 80 | dataset["tool_cost"] = data["tool_cost"] |
| 81 | dataset["token_cost"] = data["token_cost"] |
| 82 | dataset["total_cost"] = data["total_cost"] |
| 83 | dataset.to_csv(save_path, index=False) |
| 84 | return dataset |
| 85 | |
| 86 | |
| 87 | def main(args): |