Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
4ebb52af
Unverified
Commit
4ebb52af
authored
Sep 02, 2020
by
Puneetha Pai
Committed by
GitHub
Sep 02, 2020
Browse files
test_tf_common: remove un_used mixin class parameters (#6866)
parent
e71f32c0
Changes
5
Show whitespace changes
Inline
Side-by-side
Showing
5 changed files
with
0 additions
and
11 deletions
+0
-11
tests/test_modeling_tf_common.py
tests/test_modeling_tf_common.py
+0
-2
tests/test_modeling_tf_distilbert.py
tests/test_modeling_tf_distilbert.py
+0
-3
tests/test_modeling_tf_longformer.py
tests/test_modeling_tf_longformer.py
+0
-3
tests/test_modeling_tf_transfo_xl.py
tests/test_modeling_tf_transfo_xl.py
+0
-2
tests/test_modeling_tf_xlnet.py
tests/test_modeling_tf_xlnet.py
+0
-1
No files found.
tests/test_modeling_tf_common.py
View file @
4ebb52af
...
@@ -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
...
...
tests/test_modeling_tf_distilbert.py
View file @
4ebb52af
...
@@ -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
)
...
...
tests/test_modeling_tf_longformer.py
View file @
4ebb52af
...
@@ -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
=
(
(
(
...
...
tests/test_modeling_tf_transfo_xl.py
View file @
4ebb52af
...
@@ -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
):
...
...
tests/test_modeling_tf_xlnet.py
View file @
4ebb52af
...
@@ -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
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment