Fix slow tests v4.2.0 (#9561)

* Fix conversational pipeline test * LayoutLM * ProphetNet * BART * Blenderbot & small * Marian * mBART * Pegasus * Tapas tokenizer * BERT2BERT test * Style * Example requirements * TF BERT2BERT test
2021-01-13 09:55:48 -05:00
parent 04dc65e5c6
commit c949516695
12 changed files with 39 additions and 9 deletions
--- a/tests/test_modeling_tf_pegasus.py
+++ b/tests/test_modeling_tf_pegasus.py
@@ -334,6 +334,7 @@ def _long_tensor(tok_lst):

@require_sentencepiece
@require_tokenizers
+@require_tf
 class TFPegasusIntegrationTests(unittest.TestCase):
    src_text = [
        """ PG&E stated it scheduled the blackouts in response to forecasts for high winds amid dry conditions. The aim is to reduce the risk of wildfires. Nearly 800 thousand customers were scheduled to be affected by the shutoffs which were expected to last through at least midday tomorrow.""",