diff --git a/.circleci/create_circleci_config.py b/.circleci/create_circleci_config.py index 91d0dea2b4..b2b42784e4 100644 --- a/.circleci/create_circleci_config.py +++ b/.circleci/create_circleci_config.py @@ -357,9 +357,9 @@ doc_test_job = CircleCIJob( pytest_num_workers=1, ) -REGULAR_TESTS = [torch_job, tf_job, flax_job, hub_job, onnx_job, tokenization_job, processor_job, generate_job, non_model_job] # fmt: skip -EXAMPLES_TESTS = [examples_torch_job, examples_tensorflow_job] -PIPELINE_TESTS = [pipelines_torch_job, pipelines_tf_job] +REGULAR_TESTS = [torch_job, flax_job, hub_job, onnx_job, tokenization_job, processor_job, generate_job, non_model_job] # fmt: skip +EXAMPLES_TESTS = [examples_torch_job] +PIPELINE_TESTS = [pipelines_torch_job] REPO_UTIL_TESTS = [repo_utils_job] DOC_TESTS = [doc_test_job] ALL_TESTS = REGULAR_TESTS + EXAMPLES_TESTS + PIPELINE_TESTS + REPO_UTIL_TESTS + DOC_TESTS + [custom_tokenizers_job] + [exotic_models_job] # fmt: skip diff --git a/utils/tests_fetcher.py b/utils/tests_fetcher.py index f41634d612..c4b528eb89 100644 --- a/utils/tests_fetcher.py +++ b/utils/tests_fetcher.py @@ -1154,17 +1154,14 @@ def parse_commit_message(commit_message: str) -> Dict[str, bool]: JOB_TO_TEST_FILE = { - "tests_tf": r"tests/models/.*/test_modeling_tf_.*", "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_processors": r"tests/models/.*/test_(?!(?:modeling_|tokenization_)).*", # takes feature extractors, image processors, processors "examples_torch": r"examples/pytorch/.*test_.*", - "examples_tensorflow": r"examples/tensorflow/.*test_.*", "tests_exotic_models": r"tests/models/.*(?=layoutlmv|nat|deta|udop|nougat).*", "tests_custom_tokenizers": r"tests/models/.*/test_tokenization_(?=bert_japanese|openai|clip).*", # "repo_utils": r"tests/[^models].*test.*", TODO later on we might want to do - "pipelines_tf": r"tests/models/.*/test_modeling_tf_.*", "pipelines_torch": r"tests/models/.*/test_modeling_(?!(?:flax_|tf_)).*", "tests_hub": r"tests/.*", "tests_onnx": r"tests/models/.*/test_modeling_(?:tf_|(?!flax)).*",