bstraehle commited on
Commit
0d4f176
·
verified ·
1 Parent(s): 5f4828e

Update utils/utils.py

Browse files
Files changed (1) hide show
  1. utils/utils.py +5 -0
utils/utils.py CHANGED
@@ -61,6 +61,11 @@ def get_dataset(dataset_type, level):
61
  df = df[df["Level"] == level]
62
 
63
  result=[]
 
 
 
 
 
64
 
65
  # File processing
66
 
 
61
  df = df[df["Level"] == level]
62
 
63
  result=[]
64
+
65
+ for _, row in df.iterrows():
66
+ result.append([row["Question"], row["Final answer"], row["file_name"]])
67
+
68
+ return result
69
 
70
  # File processing
71