mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-04 05:10:06 +06:00
Update test reqs to include sentencepiece (#29756)
* Update test reqs * Clean
This commit is contained in:
parent
d91fd7f92c
commit
c78f57729f
1
setup.py
1
setup.py
@ -321,6 +321,7 @@ extras["testing"] = (
|
||||
"beautifulsoup4",
|
||||
"tensorboard",
|
||||
"pydantic",
|
||||
"sentencepiece",
|
||||
)
|
||||
+ extras["retrieval"]
|
||||
+ extras["modelcreation"]
|
||||
|
@ -22,7 +22,7 @@ from transformers import (
|
||||
Seq2SeqTrainingArguments,
|
||||
T5Tokenizer,
|
||||
)
|
||||
from transformers.testing_utils import TestCasePlus, require_torch, slow
|
||||
from transformers.testing_utils import TestCasePlus, require_sentencepiece, require_torch, slow
|
||||
from transformers.utils import is_datasets_available
|
||||
|
||||
|
||||
@ -30,6 +30,7 @@ if is_datasets_available():
|
||||
import datasets
|
||||
|
||||
|
||||
@require_sentencepiece
|
||||
class Seq2seqTrainerTester(TestCasePlus):
|
||||
@slow
|
||||
@require_torch
|
||||
|
Loading…
Reference in New Issue
Block a user