This commit is contained in:
Evgeny Pavlov 2024-03-20 14:28:02 -07:00 коммит произвёл GitHub
Родитель 1f12387866
Коммит a359723e41
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: B5690EEEBB952194
1 изменённых файлов: 25 добавлений и 25 удалений

Просмотреть файл

@ -405,7 +405,7 @@ def main(
)
parser.add_argument(
"--config",
default="configs/tc.prod.yml",
default="taskcluster/configs/config.prod.yml",
type=str,
help='The path to the training config. Defaults to "configs/tc.prod.yml"',
)
@ -433,21 +433,21 @@ def main(
parameters = get_taskgraph_parameters()
run_taskgraph(parsed_args.config, get_taskgraph_parameters())
match Choices[parsed_args.only] if parsed_args.only else None:
case Choices.task_group:
choice = Choices[parsed_args.only] if parsed_args.only else None
if choice == Choices.task_group:
pretty_print_task_graph()
case Choices.artifacts:
elif choice == Choices.artifacts:
pretty_print_artifacts_dir()
case Choices.training_config:
elif choice == Choices.training_config:
pretty_print_training_config(get_taskgraph_parameters())
case Choices.graph:
elif choice == Choices.graph:
serve_taskgraph_file(
parsed_args.graph_url,
parsed_args.open_graph,
parsed_args.persist_graph,
open_in_browser,
)
case None:
elif choice is None:
pretty_print_task_graph()
pretty_print_artifacts_dir()
pretty_print_training_config(parameters)