From 4ebb52afdb4dc4bcd599e7cb503763e5d4afc962 Mon Sep 17 00:00:00 2001
From: Puneetha Pai <21996583+PuneethaPai@users.noreply.github.com>
Date: Wed, 2 Sep 2020 20:24:40 +0530
Subject: [PATCH] test_tf_common: remove un_used mixin class parameters (#6866)

---
 tests/test_modeling_tf_common.py     | 2 --
 tests/test_modeling_tf_distilbert.py | 3 ---
 tests/test_modeling_tf_longformer.py | 3 ---
 tests/test_modeling_tf_transfo_xl.py | 2 --
 tests/test_modeling_tf_xlnet.py      | 1 -
 5 files changed, 11 deletions(-)

diff --git a/tests/test_modeling_tf_common.py b/tests/test_modeling_tf_common.py
index 32c79abba9..04ae52a67f 100644
--- a/tests/test_modeling_tf_common.py
+++ b/tests/test_modeling_tf_common.py
@@ -73,8 +73,6 @@ class TFModelTesterMixin:
     model_tester = None
     all_model_classes = ()
     all_generative_model_classes = ()
-    test_torchscript = True
-    test_pruning = True
     test_resize_embeddings = True
     is_encoder_decoder = False
 
diff --git a/tests/test_modeling_tf_distilbert.py b/tests/test_modeling_tf_distilbert.py
index 2c09e0fa8e..2e66334916 100644
--- a/tests/test_modeling_tf_distilbert.py
+++ b/tests/test_modeling_tf_distilbert.py
@@ -184,9 +184,6 @@ class TFDistilBertModelTest(TFModelTesterMixin, unittest.TestCase):
         if is_tf_available()
         else None
     )
-    test_pruning = True
-    test_torchscript = True
-    test_head_masking = True
 
     def setUp(self):
         self.model_tester = TFDistilBertModelTester(self)
diff --git a/tests/test_modeling_tf_longformer.py b/tests/test_modeling_tf_longformer.py
index 1282069b03..b0bd9bb260 100644
--- a/tests/test_modeling_tf_longformer.py
+++ b/tests/test_modeling_tf_longformer.py
@@ -264,9 +264,6 @@ class TFLongformerModelTester:
 
 @require_tf
 class TFLongformerModelTest(TFModelTesterMixin, unittest.TestCase):
-    test_pruning = False  # pruning is not supported
-    test_headmasking = False  # head masking is not supported
-    test_torchscript = False
 
     all_model_classes = (
         (
diff --git a/tests/test_modeling_tf_transfo_xl.py b/tests/test_modeling_tf_transfo_xl.py
index 641d1165ec..ecaad38576 100644
--- a/tests/test_modeling_tf_transfo_xl.py
+++ b/tests/test_modeling_tf_transfo_xl.py
@@ -145,8 +145,6 @@ class TFTransfoXLModelTest(TFModelTesterMixin, unittest.TestCase):
     all_model_classes = (TFTransfoXLModel, TFTransfoXLLMHeadModel) if is_tf_available() else ()
     all_generative_model_classes = () if is_tf_available() else ()
     # TODO: add this test when TFTransfoXLLMHead has a linear output layer implemented
-    test_pruning = False
-    test_torchscript = False
     test_resize_embeddings = False
 
     def setUp(self):
diff --git a/tests/test_modeling_tf_xlnet.py b/tests/test_modeling_tf_xlnet.py
index 9707686201..08e3ee1c6f 100644
--- a/tests/test_modeling_tf_xlnet.py
+++ b/tests/test_modeling_tf_xlnet.py
@@ -348,7 +348,6 @@ class TFXLNetModelTest(TFModelTesterMixin, unittest.TestCase):
     all_generative_model_classes = (
         (TFXLNetLMHeadModel,) if is_tf_available() else ()
     )  # TODO (PVP): Check other models whether language generation is also applicable
-    test_pruning = False
 
     def setUp(self):
         self.model_tester = TFXLNetModelTester(self)