This commit is contained in:
CatalinVoss 2020-11-12 10:49:33 -08:00
Родитель 396ac7fe46
Коммит 7de317cf59
1 изменённых файлов: 2 добавлений и 2 удалений

Просмотреть файл

@ -10,7 +10,7 @@ from attrdict import AttrDict
from .flags import FLAGS
from .text import levenshtein
from .io import open_remote
def pmap(fun, iterable):
pool = Pool()
@ -124,5 +124,5 @@ def save_samples_json(samples, output_path):
We set ensure_ascii=True to prevent json from escaping non-ASCII chars
in the texts.
'''
with open(output_path, 'w') as fout:
with open_remote(output_path, 'w') as fout:
json.dump(samples, fout, default=float, ensure_ascii=False, indent=2)