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", "beautifulsoup4",
"tensorboard", "tensorboard",
"pydantic", "pydantic",
"sentencepiece",
) )
+ extras["retrieval"] + extras["retrieval"]
+ extras["modelcreation"] + extras["modelcreation"]

View File

@@ -22,7 +22,7 @@ from transformers import (
Seq2SeqTrainingArguments, Seq2SeqTrainingArguments,
T5Tokenizer, 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 from transformers.utils import is_datasets_available
@@ -30,6 +30,7 @@ if is_datasets_available():
import datasets import datasets
@require_sentencepiece
class Seq2seqTrainerTester(TestCasePlus): class Seq2seqTrainerTester(TestCasePlus):
@slow @slow
@require_torch @require_torch