From 2819da02f7e3d0c0328daef12115d7a0cc78fc12 Mon Sep 17 00:00:00 2001 From: Stas Bekman Date: Wed, 18 Nov 2020 07:15:53 -0800 Subject: [PATCH] [s2s] broken test (#8613) --- examples/seq2seq/test_finetune_trainer.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/examples/seq2seq/test_finetune_trainer.py b/examples/seq2seq/test_finetune_trainer.py index 70cceae3c5..b8c0f4816c 100644 --- a/examples/seq2seq/test_finetune_trainer.py +++ b/examples/seq2seq/test_finetune_trainer.py @@ -4,7 +4,13 @@ from unittest.mock import patch from transformers import BertTokenizer, EncoderDecoderModel from transformers.file_utils import is_datasets_available -from transformers.testing_utils import TestCasePlus, execute_subprocess_async, get_gpu_count, slow +from transformers.testing_utils import ( + TestCasePlus, + execute_subprocess_async, + get_gpu_count, + require_torch_non_multi_gpu_but_fix_me, + slow, +) from transformers.trainer_callback import TrainerState from transformers.trainer_utils import set_seed @@ -46,6 +52,7 @@ class TestFinetuneTrainer(TestCasePlus): assert "test_results.json" in contents @slow + @require_torch_non_multi_gpu_but_fix_me def test_finetune_bert2bert(self): if not is_datasets_available(): return