bstraehle commited on
Commit
d907c10
·
verified ·
1 Parent(s): dc8c4f4

Update utils/utils.py

Browse files
Files changed (1) hide show
  1. utils/utils.py +2 -6
utils/utils.py CHANGED
@@ -6,16 +6,12 @@ from pptx import Presentation
6
  def get_questions(file_path, level=0):
7
  df = pd.read_json(file_path, lines=True)
8
 
9
- if level > 0:
10
- df = df[df["Level"] == level]
11
 
12
  result=[]
13
 
14
  for index, row in df.iterrows():
15
- if level > 0:
16
- result.append([row["Question"], row["Level"], row["Final answer"], row["file_name"]])
17
- else:
18
- result.append([row["Question"], "0", row["Final answer"], row["file_name"]])
19
 
20
  return result
21
 
 
6
  def get_questions(file_path, level=0):
7
  df = pd.read_json(file_path, lines=True)
8
 
9
+ df = df[df["Level"] == level]
 
10
 
11
  result=[]
12
 
13
  for index, row in df.iterrows():
14
+ result.append([row["Question"], row["Level"], row["Final answer"], row["file_name"]])
 
 
 
15
 
16
  return result
17