From df5bd86048618b7496bde0c8050bc47092a0f68c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=A1=8C=E5=97=94?= <yichang.zyc@alibaba-inc.com>
Date: Tue, 25 Oct 2022 10:15:06 +0800
Subject: [PATCH] fix a ut bug

---
 tests/trainers/test_ofa_trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/trainers/test_ofa_trainer.py b/tests/trainers/test_ofa_trainer.py
index 46dc5c8b..75b8cbbf 100644
--- a/tests/trainers/test_ofa_trainer.py
+++ b/tests/trainers/test_ofa_trainer.py
@@ -98,7 +98,7 @@ class TestOfaTrainer(unittest.TestCase):
 
         self.assertIn(ModelFile.TORCH_MODEL_BIN_FILE,
                       os.listdir(os.path.join(WORKSPACE, 'output')))
-        # shutil.rmtree(WORKSPACE)
+        shutil.rmtree(WORKSPACE)
 
 
 if __name__ == '__main__':