From 9757f362bffabe09eb15c584e24936f8245c6a36 Mon Sep 17 00:00:00 2001 From: yueqis <141804823+yueqis@users.noreply.github.com> Date: Sun, 11 Aug 2024 03:40:27 -0400 Subject: [PATCH] Update run_infer.py of gorilla to download my-languages.so (#3286) * Update run_infer.py of gorilla to download my-languages.so * add exist check, change file path, lint code --------- Co-authored-by: Graham Neubig Co-authored-by: yufansong --- evaluation/gorilla/run_infer.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/evaluation/gorilla/run_infer.py b/evaluation/gorilla/run_infer.py index d84432c3b0..44fb73c234 100644 --- a/evaluation/gorilla/run_infer.py +++ b/evaluation/gorilla/run_infer.py @@ -3,6 +3,7 @@ import json import os import pandas as pd +import requests from evaluation.gorilla.utils import encode_question, get_data_for_hub from evaluation.utils.shared import ( @@ -168,6 +169,16 @@ if __name__ == '__main__': dataset_df, output_file=output_file, eval_n_limit=args.eval_n_limit ) + file_path = os.path.join(os.path.dirname(__file__), 'my-languages.so') + # Check if the file exists + if not os.path.exists(file_path): + url = 'https://raw.githubusercontent.com/ShishirPatil/gorilla/main/eval/eval-scripts/codebleu/parser/my-languages.so' + response = requests.get(url) + with open(file_path, 'wb') as f: + f.write(response.content) + else: + print('File already exists, skipping download.') + asyncio.run( run_evaluation( dataset=dataset,