phamson02 commited on
Commit
ffa52a5
·
1 Parent(s): 9d40154
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -16,8 +16,7 @@ corpus_embeddings = pd.read_pickle("data/passage_embeds.pkl")
16
 
17
  with open("data/child_passages.tsv", "r") as f:
18
  tsv_reader = csv.reader(f, delimiter="\t")
19
- child_passage_ids = [row[0] for row in tsv_reader]
20
- child_passages = [row[1] for row in tsv_reader]
21
 
22
  with open("data/parent_passages.tsv", "r") as f:
23
  tsv_reader = csv.reader(f, delimiter="\t")
 
16
 
17
  with open("data/child_passages.tsv", "r") as f:
18
  tsv_reader = csv.reader(f, delimiter="\t")
19
+ child_passage_ids, child_passages = zip(*[(row[0], row[1]) for row in tsv_reader])
 
20
 
21
  with open("data/parent_passages.tsv", "r") as f:
22
  tsv_reader = csv.reader(f, delimiter="\t")