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

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

View File

@@ -3,7 +3,6 @@ import unittest
from collections import namedtuple
from itertools import takewhile
from tests.utils import require_torch
from transformers import (
BertTokenizer,
BertTokenizerFast,
@@ -16,6 +15,7 @@ from transformers import (
TransfoXLTokenizer,
is_torch_available,
)
from transformers.testing_utils import require_torch
from transformers.tokenization_distilbert import DistilBertTokenizerFast
from transformers.tokenization_openai import OpenAIGPTTokenizerFast
from transformers.tokenization_roberta import RobertaTokenizerFast