save ci life

huggingface · Mar 11, 2024 · 6d2ba88 · 6d2ba88
1 parent dd1c905
commit 6d2ba88
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/utils/tests_fetcher.py b/utils/tests_fetcher.py
@@ -1232,8 +1232,9 @@ def parse_commit_message(commit_message: str) -> Dict[str, bool]:
         if commit_flags["test_all"]:
             print("Force-launching all tests")
 
+        is_main_branch = not repo.head.is_detached and repo.head.ref == repo.refs.main
         diff_with_last_commit = args.diff_with_last_commit
-        if not diff_with_last_commit and not repo.head.is_detached and repo.head.ref == repo.refs.main:
+        if not diff_with_last_commit and is_main_branch:
             print("main branch detected, fetching tests against last commit.")
             diff_with_last_commit = True
 
@@ -1243,7 +1244,7 @@ def parse_commit_message(commit_message: str) -> Dict[str, bool]:
                     args.output_file,
                     diff_with_last_commit=diff_with_last_commit,
                     json_output_file=args.json_output_file,
-                    filter_models=not commit_flags["no_filter"],
+                    filter_models=(not commit_flags["no_filter"] or is_main_branch),
                 )
                 filter_tests(args.output_file, ["repo_utils"])
             except Exception as e: