measure

Author: ydshieh

.circleci/create_circleci_config.py

@@ -351,7 +351,7 @@ def job_name(self):
 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
-ALL_TESTS = [pipelines_torch_job]
+# ALL_TESTS = [pipelines_torch_job]
 
 
 def create_circleci_config(folder=None):

utils/tests_fetcher.py

@@ -1092,18 +1092,17 @@ def parse_commit_message(commit_message: str) -> dict[str, bool]:
 
 
 JOB_TO_TEST_FILE = {
-    # "tests_torch": r"tests/models/.*/test_modeling_.*",
-    # "tests_generate": r"tests/models/.*/test_modeling_.*",
-    # "tests_tokenization": r"tests/(?:models/.*/test_tokenization.*|test_tokenization_mistral_common\.py)",
-    # "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).*",
-    # "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_torch": r"tests/models/.*/test_modeling_.*",
-    # "tests_hub": r"tests/.*",
-    # "tests_non_model": r"tests/[^/]*?/test_.*\.py",
+    "tests_torch": r"tests/models/.*/test_modeling_.*",
+    "tests_generate": r"tests/models/.*/test_modeling_.*",
+    "tests_tokenization": r"tests/(?:models/.*/test_tokenization.*|test_tokenization_mistral_common\.py)",
+    "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).*",
+    "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_torch": r"tests/models/.*/test_modeling_.*",
+    "tests_hub": r"tests/.*",
+    "tests_non_model": r"tests/[^/]*?/test_.*\.py",
 }