Unverified Commit 4ebb52af authored by Puneetha Pai's avatar Puneetha Pai Committed by GitHub
Browse files

test_tf_common: remove un_used mixin class parameters (#6866)

parent e71f32c0
...@@ -73,8 +73,6 @@ class TFModelTesterMixin: ...@@ -73,8 +73,6 @@ class TFModelTesterMixin:
model_tester = None model_tester = None
all_model_classes = () all_model_classes = ()
all_generative_model_classes = () all_generative_model_classes = ()
test_torchscript = True
test_pruning = True
test_resize_embeddings = True test_resize_embeddings = True
is_encoder_decoder = False is_encoder_decoder = False
......
...@@ -184,9 +184,6 @@ class TFDistilBertModelTest(TFModelTesterMixin, unittest.TestCase): ...@@ -184,9 +184,6 @@ class TFDistilBertModelTest(TFModelTesterMixin, unittest.TestCase):
if is_tf_available() if is_tf_available()
else None else None
) )
test_pruning = True
test_torchscript = True
test_head_masking = True
def setUp(self): def setUp(self):
self.model_tester = TFDistilBertModelTester(self) self.model_tester = TFDistilBertModelTester(self)
......
...@@ -264,9 +264,6 @@ class TFLongformerModelTester: ...@@ -264,9 +264,6 @@ class TFLongformerModelTester:
@require_tf @require_tf
class TFLongformerModelTest(TFModelTesterMixin, unittest.TestCase): 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 = ( all_model_classes = (
( (
......
...@@ -145,8 +145,6 @@ class TFTransfoXLModelTest(TFModelTesterMixin, unittest.TestCase): ...@@ -145,8 +145,6 @@ class TFTransfoXLModelTest(TFModelTesterMixin, unittest.TestCase):
all_model_classes = (TFTransfoXLModel, TFTransfoXLLMHeadModel) if is_tf_available() else () all_model_classes = (TFTransfoXLModel, TFTransfoXLLMHeadModel) if is_tf_available() else ()
all_generative_model_classes = () 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 # TODO: add this test when TFTransfoXLLMHead has a linear output layer implemented
test_pruning = False
test_torchscript = False
test_resize_embeddings = False test_resize_embeddings = False
def setUp(self): def setUp(self):
......
...@@ -348,7 +348,6 @@ class TFXLNetModelTest(TFModelTesterMixin, unittest.TestCase): ...@@ -348,7 +348,6 @@ class TFXLNetModelTest(TFModelTesterMixin, unittest.TestCase):
all_generative_model_classes = ( all_generative_model_classes = (
(TFXLNetLMHeadModel,) if is_tf_available() else () (TFXLNetLMHeadModel,) if is_tf_available() else ()
) # TODO (PVP): Check other models whether language generation is also applicable ) # TODO (PVP): Check other models whether language generation is also applicable
test_pruning = False
def setUp(self): def setUp(self):
self.model_tester = TFXLNetModelTester(self) self.model_tester = TFXLNetModelTester(self)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment