wip: revert some test changes

This commit is contained in:
Julien Denize 2025-07-01 16:08:52 +02:00
parent 87e6ab5096
commit d65c8b3e2f
3 changed files with 1 additions and 3 deletions

View File

@ -66,7 +66,6 @@ NOT_DEVICE_TESTS = {
"ModelTester::test_pipeline_",
"/repo_utils/",
"/utils/",
"test_tokenization_mistral_common"
}
# allow having multiple repository checkouts and not needing to remember to rerun

View File

@ -362,7 +362,6 @@ extras["testing"] = (
"pydantic",
"sentencepiece",
"sacrebleu", # needed in trainer tests, see references to `run_translation.py`
"mistral-common[open-cv]",
)
+ extras["retrieval"]
+ extras["modelcreation"]

View File

@ -1164,7 +1164,7 @@ def parse_commit_message(commit_message: str) -> dict[str, bool]:
JOB_TO_TEST_FILE = {
"tests_torch": r"tests/models/.*/test_modeling_(?!(?:flax_|tf_)).*",
"tests_generate": r"tests/models/.*/test_modeling_(?!(?:flax_|tf_)).*",
"tests_tokenization": r"tests/(models/.*/|)test_tokenization.*",
"tests_tokenization": r"tests/models/.*/test_tokenization.*",
"tests_processors": r"tests/models/.*/test_(?!(?:modeling_|tokenization_)).*", # takes feature extractors, image processors, processors
"examples_torch": r"examples/pytorch/.*test_.*",
"tests_exotic_models": r"tests/models/.*(?=layoutlmv|nat|deta|udop|nougat).*",