The extended trainer tests should require torch (#12650)

This commit is contained in:
Lysandre Debut 2021-07-12 15:47:05 +02:00 committed by GitHub
parent 0af8579bbe
commit fb5665b5ad
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions

View File

@ -28,6 +28,7 @@ from transformers.testing_utils import (
execute_subprocess_async,
get_gpu_count,
get_torch_dist_unique_port,
require_torch,
require_torch_gpu,
require_torch_multi_gpu,
require_torch_non_multi_gpu,
@ -69,6 +70,7 @@ def require_apex(test_case):
return test_case
@require_torch
class TestTrainerExt(TestCasePlus):
def run_seq2seq_quick(
self,