riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-06 14:20:04 +06:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

Update test reqs to include sentencepiece (#29756)

Browse Source 
* 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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
setup.py
View File

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

3
tests/trainer/test_trainer_seq2seq.py
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