Fix compatibility (#480)
This commit is contained in:
Родитель
1f12387866
Коммит
a359723e41
|
@ -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__":
|
||||||
|
|
Загрузка…
Ссылка в новой задаче