fix a ut bug

This commit is contained in:
行嗔
2022-10-25 10:15:06 +08:00
parent 4276c5434f
commit df5bd86048

View File

@@ -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__':