Update helper.py

helper.py

@@ -1,15 +1,13 @@
 import pandas as pd
 
-LEVEL = 1
+LEVEL=1
 
-FILE_NAME = "data/gaia_20.jsonl"
-#FILE_NAME = "data/gaia_validation.jsonl"
+FILE_NAME="data/gaia_validation_20.jsonl"
 
 def get_questions():
-    df = pd.read_json(FILE_NAME, lines = True)
-    level_df = df[df["Level"] == LEVEL]
-
-    result = []
+    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"]])