diff --git a/data/unified_datasets/sgd/preprocess.py b/data/unified_datasets/sgd/preprocess.py index 255e8008787f90593f77b333ca391adf379ef445..099d26e3a8bb34a681c0ae0590134b1848dad626 100644 --- a/data/unified_datasets/sgd/preprocess.py +++ b/data/unified_datasets/sgd/preprocess.py @@ -353,8 +353,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, ontology if __name__ == '__main__':