grady / helper.py
bstraehle's picture
Update helper.py
0026068 verified
raw
history blame
314 Bytes
import pandas as pd
def get_questions(level):
df = pd.read_json("gaia.jsonl", 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