use dictionary comprehension to pythonic code

avx_docs
Luo Tao 9 years ago
parent dc420f90f0
commit 11b7625d48

@ -25,11 +25,17 @@ def hook(settings, src_dict_path, trg_dict_path, is_generating, file_list,
# job_mode = 0: generating mode
settings.job_mode = not is_generating
settings.src_dict = dict()
for line_count, line in enumerate(open(src_dict_path, "r")):
settings.src_dict[line.strip()] = line_count
with open(src_dict_path, "r") as fin:
settings.src_dict = {
line.strip(): line_count
for line_count, line in enumerate(fin)
}
settings.trg_dict = dict()
for line_count, line in enumerate(open(trg_dict_path, "r")):
settings.trg_dict[line.strip()] = line_count
with open(trg_dict_path, "r") as fin:
settings.trg_dict = {
line.strip(): line_count
for line_count, line in enumerate(fin)
}
settings.logger.info("src dict len : %d" % (len(settings.src_dict)))
settings.sample_count = 0

Loading…
Cancel
Save