Skip to content

Commit

Permalink
linting
Browse files Browse the repository at this point in the history
  • Loading branch information
rishsriv committed May 30, 2024
1 parent d3ec84f commit a13e720
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions eval/api_runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -270,7 +270,7 @@ def run_api_eval(args):
)

output_df = pd.DataFrame(output_rows)

print(output_df.groupby("query_category")[["correct", "error_db_exec"]].mean())
output_df = output_df.sort_values(by=["db_name", "query_category", "question"])
# get directory of output_file and create if not exist
Expand All @@ -288,7 +288,7 @@ def run_api_eval(args):
"w",
) as f:
json.dump(results, f)

del output_df["prompt"]
try:
output_df.to_csv(output_file, index=False, float_format="%.2f")
Expand Down

0 comments on commit a13e720

Please sign in to comment.