Browse Source

fix a ut bug

master
行嗔 2 years ago
parent
commit
df5bd86048
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      tests/trainers/test_ofa_trainer.py

+ 1
- 1
tests/trainers/test_ofa_trainer.py 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__':


Loading…
Cancel
Save