Update utils/utils.py
Browse files- 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 |
|