SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

test_tf_common: remove un_used mixin class parameters (#6866)

Browse Source
This commit is contained in:
Puneetha Pai
2020-09-02 20:24:40 +05:30
committed by GitHub
parent e71f32c0ef
commit 4ebb52afdb
5 changed files with 0 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_modeling_tf_common.py
View File

@@ -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

3
tests/test_modeling_tf_distilbert.py
View File

@@ -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)

3
tests/test_modeling_tf_longformer.py
View File

@@ -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 = (
( (

2
tests/test_modeling_tf_transfo_xl.py
View File

@@ -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):

1
tests/test_modeling_tf_xlnet.py
View File

@@ -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)