bstraehle commited on
Commit
3dece43
·
verified ·
1 Parent(s): ca8ba7d

Update utils/utils.py

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