Enable more test_torchscript (#16679)

* update _create_and_check_torchscript * Enable test_torchscript * clear_class_registry Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2022-04-11 18:23:35 +02:00
parent 3918d6a9d6
commit c04619ecf3
32 changed files with 9 additions and 39 deletions
--- a/tests/data2vec/test_modeling_data2vec_audio.py
+++ b/tests/data2vec/test_modeling_data2vec_audio.py
@@ -372,7 +372,6 @@ class Data2VecAudioModelTest(ModelTesterMixin, unittest.TestCase):
    )
    test_pruning = False
    test_headmasking = False
-    test_torchscript = False

    def setUp(self):
        self.model_tester = Data2VecAudioModelTester(self)