Update test reqs to include sentencepiece (#29756)

* Update test reqs

* Clean
This commit is contained in:
Zach Mueller 2024-03-20 11:53:42 -04:00 committed by GitHub
parent d91fd7f92c
commit c78f57729f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 1 deletions

View File

@ -321,6 +321,7 @@ extras["testing"] = (
"beautifulsoup4",
"tensorboard",
"pydantic",
"sentencepiece",
)
+ extras["retrieval"]
+ extras["modelcreation"]

View File

@ -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