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

Move tests/utils.py -> transformers/testing_utils.py (#5350)

Browse Source
This commit is contained in:
Sam Shleifer
2020-07-01 10:31:17 -04:00
committed by GitHub
parent 9c219305f5
commit 13deb95a40
66 changed files with 66 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_pipelines.py
View File

@@ -3,8 +3,7 @@ from typing import Iterable, List, Optional
from transformers import pipeline
from transformers.pipelines import SUPPORTED_TASKS, DefaultArgumentHandler, Pipeline
from .utils import require_tf, require_torch, slow, torch_device
from transformers.testing_utils import require_tf, require_torch, slow, torch_device
DEFAULT_DEVICE_NUM = -1 if torch_device == "cpu" else 0