setup.py

@@ -5,7 +5,7 @@ REQUIREMENTS = [
     'absl-py==0.9.0',
     'allennlp==1.2.0',
     'conllu==2.3.2',
-    'dataclasses==0.5',
+    'dataclasses;python_version<"3.7"',
     'dataclasses-json==0.5.2',
     'joblib==0.14.1',
     'jsonnet==0.15.0',

tests/test_predict.py

 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."