Unverified Commit c78f5772 authored by Zach Mueller's avatar Zach Mueller Committed by GitHub
Browse files

Update test reqs to include sentencepiece (#29756)

* Update test reqs

* Clean
parent d91fd7f9
......@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment