"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "c22545aa407c6a51c18a811677411f18fa4a32d5"
Unverified Commit 2819da02 authored by Stas Bekman's avatar Stas Bekman Committed by GitHub
Browse files

[s2s] broken test (#8613)

parent 9fa3ed1a
...@@ -4,7 +4,13 @@ from unittest.mock import patch ...@@ -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): ...@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment