From 92f02dbe3a6e269e6626355d53ecd6770a6740c4 Mon Sep 17 00:00:00 2001 From: zqwerty <zhuq96@hotmail.com> Date: Sun, 30 Jan 2022 17:38:49 +0800 Subject: [PATCH] update multiwoz21 preprocess --- data/unified_datasets/multiwoz21/preprocess.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/data/unified_datasets/multiwoz21/preprocess.py b/data/unified_datasets/multiwoz21/preprocess.py index d870026a..14032081 100644 --- a/data/unified_datasets/multiwoz21/preprocess.py +++ b/data/unified_datasets/multiwoz21/preprocess.py @@ -889,8 +889,8 @@ def preprocess(): with ZipFile('data.zip', 'w', ZIP_DEFLATED) as zf: for filename in os.listdir(new_data_dir): zf.write(f'{new_data_dir}/{filename}') - # rmtree(original_data_dir) - # rmtree(new_data_dir) + rmtree(original_data_dir) + rmtree(new_data_dir) return dialogues, init_ontology if __name__ == '__main__': -- GitLab