diff --git a/tests/test_predict.py b/tests/test_predict.py index 42bc4936ad627971fe7a07f9c405fead8590633f..2a56bd9baff30a6d34d3ad6bb16ce4ccdea71792 100644 --- a/tests/test_predict.py +++ b/tests/test_predict.py @@ -1,8 +1,6 @@ import os import pathlib -import shutil import unittest -from unittest import mock import combo.data as data import combo.predict as predict @@ -14,16 +12,6 @@ class PredictionTest(unittest.TestCase): TESTS_ROOT = PROJECT_ROOT / "tests" FIXTURES_ROOT = TESTS_ROOT / "fixtures" - def setUp(self) -> None: - def _cleanup_archive_dir_without_logging(path: str): - if os.path.exists(path): - shutil.rmtree(path) - - self.patcher = mock.patch( - "allennlp.models.archival._cleanup_archive_dir", _cleanup_archive_dir_without_logging - ) - self.mock_cleanup_archive_dir = self.patcher.start() - def test_prediction_are_equal_given_the_same_input_in_different_form(self): # given raw_sentence = "Test."