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