From f5b50c6b8ecf8fd69922ac916b860a9ca21d466c Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Tue, 25 Feb 2020 16:41:54 +0100 Subject: [PATCH] make style --- tests/test_modeling_gpt2.py | 1 - tests/test_modeling_transfo_xl.py | 1 - tests/test_modeling_xlm.py | 1 - tests/test_modeling_xlnet.py | 1 - 4 files changed, 4 deletions(-) diff --git a/tests/test_modeling_gpt2.py b/tests/test_modeling_gpt2.py index 0ec1be4812..fb2474ca56 100644 --- a/tests/test_modeling_gpt2.py +++ b/tests/test_modeling_gpt2.py @@ -264,7 +264,6 @@ class GPT2ModelTest(ModelTesterMixin, unittest.TestCase): class GPT2ModelLanguageGenerationTest(unittest.TestCase): - @slow def test_lm_generate_gpt2(self): model = GPT2LMHeadModel.from_pretrained("gpt2") diff --git a/tests/test_modeling_transfo_xl.py b/tests/test_modeling_transfo_xl.py index f59ed628eb..6ba12e4738 100644 --- a/tests/test_modeling_transfo_xl.py +++ b/tests/test_modeling_transfo_xl.py @@ -215,7 +215,6 @@ class TransfoXLModelTest(ModelTesterMixin, unittest.TestCase): class TransfoXLModelLanguageGenerationTest(unittest.TestCase): - @slow def test_lm_generate_transfo_xl_wt103(self): model = TransfoXLLMHeadModel.from_pretrained("transfo-xl-wt103") diff --git a/tests/test_modeling_xlm.py b/tests/test_modeling_xlm.py index bf7d7b485b..a00af4f726 100644 --- a/tests/test_modeling_xlm.py +++ b/tests/test_modeling_xlm.py @@ -400,7 +400,6 @@ class XLMModelTest(ModelTesterMixin, unittest.TestCase): class XLMModelLanguageGenerationTest(unittest.TestCase): - @slow def test_lm_generate_xlm_mlm_en_2048(self): model = XLMWithLMHeadModel.from_pretrained("xlm-mlm-en-2048") diff --git a/tests/test_modeling_xlnet.py b/tests/test_modeling_xlnet.py index 160da136e5..15636fc4a3 100644 --- a/tests/test_modeling_xlnet.py +++ b/tests/test_modeling_xlnet.py @@ -514,7 +514,6 @@ class XLNetModelTest(ModelTesterMixin, unittest.TestCase): class XLNetModelLanguageGenerationTest(unittest.TestCase): - @slow def test_lm_generate_xlnet_base_cased(self): model = XLNetLMHeadModel.from_pretrained("xlnet-base-cased")