From eb6db3741361e68f4ddf506098629c710f429920 Mon Sep 17 00:00:00 2001 From: Hanna Moazam Date: Sat, 22 Jun 2024 18:30:01 +0100 Subject: [PATCH] Added trust_remote_code=True to hotpot_qa dataset to skip prompt for allowing custom code to be run --- dspy/datasets/hotpotqa.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dspy/datasets/hotpotqa.py b/dspy/datasets/hotpotqa.py index 04684ac57e..89d5ed221d 100644 --- a/dspy/datasets/hotpotqa.py +++ b/dspy/datasets/hotpotqa.py @@ -11,8 +11,8 @@ def __init__(self, *args, only_hard_examples=True, keep_details='dev_titles', un assert only_hard_examples, "Care must be taken when adding support for easy examples." \ "Dev must be all hard to match official dev, but training can be flexible." - hf_official_train = load_dataset("hotpot_qa", 'fullwiki', split='train') - hf_official_dev = load_dataset("hotpot_qa", 'fullwiki', split='validation') + hf_official_train = load_dataset("hotpot_qa", 'fullwiki', split='train', trust_remote_code=True) + hf_official_dev = load_dataset("hotpot_qa", 'fullwiki', split='validation', trust_remote_code=True) official_train = [] for raw_example in hf_official_train: