File size: 383 Bytes
da9fcee
 
c8f728b
b1e959c
c8f728b
 
 
 
b1e959c
c8f728b
da9fcee
4cd6fae
 
0026068
0fccc5c
4cd6fae
0026068
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
import pandas as pd

LEVEL = 1

FILE_NAME = "gaia_20.jsonl"
#FILE_NAME = "gaia_validation.jsonl"

def get_questions():
    df = pd.read_json(FILE_NAME, lines = True)
    level_df = df[df["Level"] == LEVEL]

    result = []
    
    for index, row in level_df.iterrows():
        result.append([row["Level"], row["Question"], row["file_name"], row["Final answer"]])

    return result