SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

The extended trainer tests should require torch (#12650)

Browse Source
This commit is contained in:
Lysandre Debut
2021-07-12 15:47:05 +02:00
committed by GitHub
parent 0af8579bbe
commit fb5665b5ad
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/extended/test_trainer_ext.py
View File

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