Reduce the time spent for the TF slow tests (#10152)

* rework savedmodel slow test

* Improve savedmodel tests

* Remove useless content
This commit is contained in:
Julien Plu
2021-02-18 15:52:57 +01:00
committed by GitHub
parent 14ed3b978e
commit 2acae50a0c
7 changed files with 91 additions and 166 deletions

View File

@@ -370,27 +370,10 @@ class TFLEDModelTest(TFModelTesterMixin, unittest.TestCase):
# TODO JP: Make LED XLA compliant
pass
def test_saved_model_with_attentions_output(self):
# Temporarily disable this test in order to find
# how to better handle it without timing out the CI
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."""