diff --git a/.github/triage/jax_toolbox_triage/args.py b/.github/triage/jax_toolbox_triage/args.py index de248cef9..7946bcead 100644 --- a/.github/triage/jax_toolbox_triage/args.py +++ b/.github/triage/jax_toolbox_triage/args.py @@ -260,6 +260,11 @@ def parse_args(args=None) -> argparse.Namespace: jax:https://@host/repo.git,xla:https://@host/repo.git """, ) + version_search_args.add_argument( + "--exclude-transformer-engine", + action="store_true", + help="Exclude transformer-engine from the list of optional software to triage.", + ) parser.add_argument( "-v", "--container-mount", @@ -375,4 +380,7 @@ def parse_args(args=None) -> argparse.Namespace: "--container must be passed for the container-level search" ) + if args.exclude_transformer_engine: + optional_software.remove("transformer-engine") + return args