Skip to content

GitLab

Unverified Commit a7ae2b5b authored by Dongruixuan Li's avatar Dongruixuan Li Committed by GitHub
Browse files

[eval-hotfix] set few_shot_data to None when few shot is disabled (#5422)

parent 049121d1
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
applications/ColossalEval/colossal_eval/dataset/agieval.py
View file @ a7ae2b5b
...@@ -201,7 +201,7 @@ class AGIEvalDataset(BaseDataset): ...@@ -201,7 +201,7 @@ class AGIEvalDataset(BaseDataset):
for file in files: for file in files:
dataset_name = os.path.basename(file)[0 : -len(".jsonl")] dataset_name = os.path.basename(file)[0 : -len(".jsonl")]
few_shot_data = [] few_shot_data = None
if few_shot: if few_shot:
# process demo once if it is few-shot-CoT # process demo once if it is few-shot-CoT
few_shot_data = combine_prompt(prompt_path, dataset_name, load_explanation=False, chat_mode=False) few_shot_data = combine_prompt(prompt_path, dataset_name, load_explanation=False, chat_mode=False)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment