From 0ef21cf8aa835f90249e384713f05a093864deae Mon Sep 17 00:00:00 2001
From: zqwerty <zhuq96@hotmail.com>
Date: Fri, 6 May 2022 17:13:29 +0800
Subject: [PATCH] update dailydialog preprocess.py
---
data/unified_datasets/dailydialog/preprocess.py | 8 +++-----
1 file changed, 3 insertions(+), 5 deletions(-)
diff --git a/data/unified_datasets/dailydialog/preprocess.py b/data/unified_datasets/dailydialog/preprocess.py
index e04a1587..8a04aae5 100644
--- a/data/unified_datasets/dailydialog/preprocess.py
+++ b/data/unified_datasets/dailydialog/preprocess.py
@@ -147,12 +147,10 @@ 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, ontology
-
-
if __name__ == '__main__':
- preprocess()
\ No newline at end of file
+ preprocess()
--
GitLab