Lines Matching +full:pytorchbot +full:- +full:token
8 # And re-run the test locally with ones PAT
54 ) -> Any:
57 def get_mocked_queries() -> Any:
60 with gzip.open(gql_db_fname, encoding="utf-8", mode="rt") as f:
63 def save_mocked_queries(obj: Any) -> None:
64 with gzip.open(gql_db_fname, encoding="utf-8", mode="wt") as f:
80 err_msg += " GitHub Personal Access Token passed via GITHUB_TOKEN,"
101 def mocked_gh_graphql(query: str, **kwargs: Any) -> Any:
102 def key_function(query: str, kwargs: Any) -> str:
103 return f"query_sha={sha256(query.encode('utf-8')).hexdigest()} " + " ".join(
107 def gh_graphql_wrapper(query: str, kwargs: Any) -> Any:
113 def mocked_rockset_results(head_sha: str, merge_base: str, num_retries: int = 3) -> Any:
123 def mocked_drci_classifications(pr_num: int, project: str, num_retries: int = 3) -> Any:
133 def mock_parse_args(revert: bool = False, force: bool = False) -> Any:
135 def __init__(self) -> None:
150 ) -> None:
161 ) -> None:
174 ) -> None:
178 def mock_gh_get_info() -> Any:
190 def mocked_read_merge_rules_NE(repo: Any, org: str, project: str) -> List[MergeRule]:
202 def mocked_read_merge_rules(repo: Any, org: str, project: str) -> List[MergeRule]:
210 "pull / linux-xenial-cuda11.3-py3.7-gcc7 / build",
217 approved_by=["pytorchbot"],
221 "pull / linux-focal-py3_8-clang9-xla / build",
222 "pull / linux-focal-py3_8-clang9-xla / test (xla, 1, 1, linux.12xlarge)",
231 ) -> List[MergeRule]:
254 def mocked_read_merge_rules_raise(repo: Any, org: str, project: str) -> List[MergeRule]:
258 def xla_merge_rules(repo: Any, org: str, project: str) -> List[MergeRule]:
261 name=" OSS CI / pytorchbot / XLA",
263 approved_by=["pytorchbot"],
267 "pull / linux-bionic-py3_8-clang8-xla / build",
268 "pull / linux-bionic-py3_8-clang8-xla / test (xla, 1, 1, linux.4xlarge)",
269 …"inductor / cuda11.8-py3.10-gcc7-sm86 / test (inductor_torchbench_dynamic, 1, 1, linux.g5.4xlarge.…
276 def empty_rockset_results(head_sha: str, merge_base: str) -> List[Dict[str, Any]]:
281 def __init__(self) -> None:
284 def commits_resolving_gh_pr(self, pr_num: int) -> List[str]:
287 def commit_message(self, ref: str) -> str:
297 def test_merge_rules_valid(self, *args: Any) -> None:
304 def test_match_rules(self, *args: Any) -> None:
311 def test_read_merge_rules_fails(self, *args: Any) -> None:
322 def test_match_rules_approvers(self, *args: Any) -> None:
341 def test_lint_fails(self, *args: Any) -> None:
347 def test_get_last_comment(self, *args: Any) -> None:
351 self.assertEqual(comment.author_login, "github-actions")
355 def test_get_author_null(self, *args: Any) -> None:
371 def test_large_diff(self, *args: Any) -> None:
378 def test_internal_changes(self, *args: Any) -> None:
383 def test_comments_pagination(self, *args: Any) -> None:
388 def test_gql_complexity(self, *args: Any) -> None:
400 def test_gql_retrieve_checksuites(self, *args: Any) -> None:
405 def test_team_members(self, *args: Any) -> None:
407 dev_infra_team = gh_get_team_members("pytorch", "pytorch-dev-infra")
413 def test_get_author_many_commits(self, *args: Any) -> None:
422 def test_pending_status_check(self, *args: Any) -> None:
432 def test_get_author_many_reviews(self, *args: Any) -> None:
440 def get_co_authors(self, *args: Any) -> None:
441 """Tests that co-authors are recognized"""
445 self.assertIn("Co-authored-by:", pr.gen_commit_message())
447 def test_get_checkruns_many_runs(self, *args: Any) -> None:
453 "pull / linux-docs / build-docs-cpp-false" in conclusions.keys()
456 def test_cancelled_gets_ignored(self, *args: Any) -> None:
466 def test_get_review_comment_by_id(self, *args: Any) -> None:
476 def test_main_revert(self, mock_revert: Any, *args: Any) -> None:
486 ) -> None:
501 def test_main_merge(self, mock_merge: Any, *args: Any) -> None:
513 def test_revert_rules(self, *args: Any) -> None:
519 def test_get_changed_files(self, *args: Any) -> None:
531 def test_revert_codev_abandoned_diff_succeeds(self, *args: Any) -> None:
535 def commit_message(self, ref: str) -> str:
541 def test_pr_changed_submodule_detection(self, *args: Any) -> None:
542 # Updates submodule during dev-cycle but reverts it later
557 def test_remove_job_name_suffix(self, *args: Any) -> None:
560 …"name": "linux-bionic-cuda12.1-py3.10-gcc9-sm86 / test (default, 1, 5, linux.g5.4xlarge.nvidia.gpu…
561 "expected": "linux-bionic-cuda12.1-py3.10-gcc9-sm86 / test (default)",
564 … "name": "android-emulator-build-test / build-and-test (default, 1, 1, ubuntu-20.04-16x)",
565 "expected": "android-emulator-build-test / build-and-test (default)",
568 "name": "linux-focal-rocm5.4.2-py3.8 / build",
569 "expected": "linux-focal-rocm5.4.2-py3.8 / build",
572 "name": "libtorch-cpu-shared-with-deps-release-build",
573 "expected": "libtorch-cpu-shared-with-deps-release-build",
576 "name": "manywheel-py3_8-cuda11_8-test / test",
577 "expected": "manywheel-py3_8-cuda11_8-test / test",
580 "name": "lintrunner / linux-job",
581 "expected": "lintrunner / linux-job",
592 def test_get_merge_base(self, *args: Any) -> None:
595 mock_merge_base = "mocked-sha"
614 def test_get_classifications(self, *args: Any) -> None:
625 "pull / linux-focal-py3.11-clang10 / test (dynamo, 1, 2, linux.2xlarge)"
631 "trunk / win-vs2019-cpu-py3 / test (default, 2, 3, windows.4xlarge.nonephemeral)"
637 "pull / linux-jammy-py3.8-gcc11 / test (distributed, 1, 2, linux.2xlarge)"
643 … "pull / linux-focal-cuda11.8-py3.10-gcc9 / test (distributed, 1, 3, linux.8xlarge.nvidia.gpu)"
657 # Not set any threshold, defaults to -1 to ignore all flaky and broken trunk failures
682 def test_get_classifications_flaky_fullname(self, *args: Any) -> None:
696 def test_get_classifications_invalid_cancel(self, *args: Any) -> None:
712 def test_get_classifications_similar_failures(self, *args: Any) -> None:
726 def test_get_classifications_unstable(self, *args: Any) -> None:
735 workflow_name = "linux-bionic-cuda12.1-py3.10-gcc9-bazel-test"
736 job_name = "build-and-test (default, 1, 1, linux.4xlarge.nvidia.gpu, unstable)"
758 workflow_name = "test-llama-app"
759 job_name = "mobile-job (android)"
771 def test_get_classifications_broken_trunk(self, *args: Any) -> None:
832 def test_ignore_current(self, *args: Any) -> None:
837 …flaky = "pull / linux-focal-cuda11.8-py3.10-gcc9 / test (distributed, 1, 3, linux.8xlarge.nvidia.g…
839 "pull / linux-focal-py3.11-clang10 / test (dynamo, 1, 2, linux.2xlarge)"
862 def test_get_classifications_wrong_workflow_name(self, *args: Any) -> None:
866 check_name = "linux-binary-conda / conda-py3_8-cuda11_8-build / build"
867 …name_workflow_path = ".github/workflows/generated-linux-binary-conda-nightly.yml / conda-py3_8-cud…
897 def test_ignore_failures_older_run_same_workflow(self, *args: Any) -> None:
916 def test_dont_ignore_flaky_failures(self, *args: Any) -> None:
918 Regression test for https://github.com/pytorch/test-infra/issues/4126
937 def test_get_classifications(self, *args: Any) -> None:
952 def test_get_classifications_drci_checkrun_not_found(self, *args: Any) -> None:
1018 def test_pr_dependencies(self, *args: Any) -> None:
1028 def test_pr_dependencies_ghstack(self, *args: Any) -> None:
1054 ) -> None:
1094 "### Background: Gradient Pre-Divide"