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"] = ( ...@@ -321,6 +321,7 @@ extras["testing"] = (
"beautifulsoup4", "beautifulsoup4",
"tensorboard", "tensorboard",
"pydantic", "pydantic",
"sentencepiece",
) )
+ extras["retrieval"] + extras["retrieval"]
+ extras["modelcreation"] + extras["modelcreation"]
......
...@@ -22,7 +22,7 @@ from transformers import ( ...@@ -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(): ...@@ -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
......
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