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