Skip to content

GitLab

Unverified Commit fb5665b5 authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

The extended trainer tests should require torch (#12650)

parent 0af8579b
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
tests/extended/test_trainer_ext.py
View file @ fb5665b5
...@@ -28,6 +28,7 @@ from transformers.testing_utils import ( ...@@ -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): ...@@ -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,
......
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