[s2s] broken test (#8613)
This commit is contained in:
@@ -4,7 +4,13 @@ from unittest.mock import patch
|
|||||||
|
|
||||||
from transformers import BertTokenizer, EncoderDecoderModel
|
from transformers import BertTokenizer, EncoderDecoderModel
|
||||||
from transformers.file_utils import is_datasets_available
|
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_callback import TrainerState
|
||||||
from transformers.trainer_utils import set_seed
|
from transformers.trainer_utils import set_seed
|
||||||
|
|
||||||
@@ -46,6 +52,7 @@ class TestFinetuneTrainer(TestCasePlus):
|
|||||||
assert "test_results.json" in contents
|
assert "test_results.json" in contents
|
||||||
|
|
||||||
@slow
|
@slow
|
||||||
|
@require_torch_non_multi_gpu_but_fix_me
|
||||||
def test_finetune_bert2bert(self):
|
def test_finetune_bert2bert(self):
|
||||||
if not is_datasets_available():
|
if not is_datasets_available():
|
||||||
return
|
return
|
||||||
|
|||||||
Reference in New Issue
Block a user