bstraehle commited on
Commit
66cdcb0
·
verified ·
1 Parent(s): a3e5b28

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -1
utils.py CHANGED
@@ -11,7 +11,8 @@ def get_questions(file_path, level):
11
  ###
12
  token = os.getenv("HF_TOKEN")
13
  login(token)
14
- df = load_dataset("gaia-benchmark/GAIA", "2023_all", split="validation", trust_remote_code=True)
 
15
  ###
16
 
17
  if level > 0:
 
11
  ###
12
  token = os.getenv("HF_TOKEN")
13
  login(token)
14
+ dataset = load_dataset("gaia-benchmark/GAIA", "2023_all", split="validation", trust_remote_code=True)
15
+ df = pd.DataFrame(dataset)
16
  ###
17
 
18
  if level > 0: