riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Disable temporarily too slow tests (Longformer/LED) (#10062)

Browse Source 
* Disable temporarily too slow tests

* Fix style

* Fix template
This commit is contained in:
Julien Plu 2021-02-08 12:32:31 +01:00 committed by GitHub
parent b1aa4982cd
commit 8bb52bd240
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 28 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
tests/test_modeling_tf_led.py
View File

@ -353,10 +353,6 @@ class TFLEDModelTest(TFModelTesterMixin, unittest.TestCase):
self.assertEqual(model.config.output_hidden_states, True)
check_encoder_attentions_output(outputs)
def test_saved_model_creation(self):
# This test is too long (>30sec) and makes fail the CI
pass
def test_mixed_precision(self):
# TODO JP: Make LED float16 compliant
pass
@ -377,6 +373,22 @@ class TFLEDModelTest(TFModelTesterMixin, unittest.TestCase):
# Need to check with PVP how to properly fix this
pass
@slow
def test_saved_model_with_hidden_states_output(self):
# Temporarily disable this test in order to find
# how to better handle it without timing out the CI
pass
def test_saved_model_creation(self):
# This test is too long (>30sec) and makes fail the CI
pass
@slow
def test_saved_model_creation_extended(self):
# Temporarily disable this test in order to find
# how to better handle it without timing out the CI
pass
def _assert_tensors_equal(a, b, atol=1e-12, prefix=""):
"""If tensors not close, or a and b arent both tensors, raise a nice Assertion error."""

12
tests/test_modeling_tf_longformer.py
View File

@ -351,10 +351,22 @@ class TFLongformerModelTest(TFModelTesterMixin, unittest.TestCase):
# Need to check with PVP how to properly fix this
pass
@slow
def test_saved_model_with_hidden_states_output(self):
# Temporarily disable this test in order to find
# how to better handle it without timing out the CI
pass
def test_saved_model_creation(self):
# This test is too long (>30sec) and makes fail the CI
pass
@slow
def test_saved_model_creation_extended(self):
# Temporarily disable this test in order to find
# how to better handle it without timing out the CI
pass
def test_mixed_precision(self):
# TODO JP: Make Longformer float16 compliant
pass