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

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

Browse Source 
* 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
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/test_modeling_tf_longformer.py
View File

@@ -339,28 +339,10 @@ class TFLongformerModelTest(TFModelTesterMixin, unittest.TestCase):
config_and_inputs = self.model_tester.prepare_config_and_inputs()
self.model_tester.create_and_check_for_multiple_choice(*config_and_inputs)
@slow
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 test_mixed_precision(self):
# TODO JP: Make Longformer float16 compliant
pass