Skip to content

Commit 11b7625

Browse files
committed
use dictionary comprehension to pythonic code
1 parent dc420f9 commit 11b7625

File tree

1 file changed

+10
-4
lines changed

1 file changed

+10
-4
lines changed

demo/seqToseq/dataprovider.py

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,11 +25,17 @@ def hook(settings, src_dict_path, trg_dict_path, is_generating, file_list,
2525
# job_mode = 0: generating mode
2626
settings.job_mode = not is_generating
2727
settings.src_dict = dict()
28-
for line_count, line in enumerate(open(src_dict_path, "r")):
29-
settings.src_dict[line.strip()] = line_count
28+
with open(src_dict_path, "r") as fin:
29+
settings.src_dict = {
30+
line.strip(): line_count
31+
for line_count, line in enumerate(fin)
32+
}
3033
settings.trg_dict = dict()
31-
for line_count, line in enumerate(open(trg_dict_path, "r")):
32-
settings.trg_dict[line.strip()] = line_count
34+
with open(trg_dict_path, "r") as fin:
35+
settings.trg_dict = {
36+
line.strip(): line_count
37+
for line_count, line in enumerate(fin)
38+
}
3339

3440
settings.logger.info("src dict len : %d" % (len(settings.src_dict)))
3541
settings.sample_count = 0

0 commit comments

Comments
 (0)