diff --git a/source/tests/pt/test_training.py b/source/tests/pt/test_training.py index c7094712ad..2926465775 100644 --- a/source/tests/pt/test_training.py +++ b/source/tests/pt/test_training.py @@ -66,8 +66,6 @@ def test_dp_train(self): ) trainer_finetune_empty.run() - self.tearDown() - def test_trainable(self): fix_params = deepcopy(self.config) fix_params["model"]["descriptor"]["trainable"] = False @@ -96,8 +94,6 @@ def test_trainable(self): model_dict_before_training[key], model_dict_after_training[key] ) - self.tearDown() - def tearDown(self): for f in os.listdir("."): if f.startswith("model") and f.endswith(".pt"):