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