diff --git a/assets/ar/factuality_disinformation_harmful_content/propaganda/ArAIEval23T1A_FewShot.py b/assets/ar/factuality_disinformation_harmful_content/propaganda/ArAIEval23T1A_FewShot.py index feccb1b1..17e59826 100644 --- a/assets/ar/factuality_disinformation_harmful_content/propaganda/ArAIEval23T1A_FewShot.py +++ b/assets/ar/factuality_disinformation_harmful_content/propaganda/ArAIEval23T1A_FewShot.py @@ -22,7 +22,7 @@ def config(): "model_args": { "max_tries": 3, }, - "general_args": {"test_split": "task1A", "train_split":"task1A"}, + "general_args": {"test_split": "task1A", "train_split": "task1A"}, } diff --git a/llmebench/utils.py b/llmebench/utils.py index 19df772b..8124bab3 100644 --- a/llmebench/utils.py +++ b/llmebench/utils.py @@ -80,7 +80,6 @@ def get_data_paths(config, split): dataset_args = config.get("dataset_args", {}) dataset = config["dataset"](**dataset_args) - if split == "test": data_args = config.get("general_args", {}) elif split == "train":