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:
@@ -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."""
|
||||
|
||||
Reference in New Issue
Block a user