Fix compatibility (#480)

mozilla · Mar 20, 2024 · a359723 · a359723
1 parent 1f12387
commit a359723
Showing 1 changed file with 25 additions and 25 deletions.
diff --git a/utils/preflight_check.py b/utils/preflight_check.py
@@ -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,30 +433,30 @@ 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:
-            pretty_print_task_graph()
-        case Choices.artifacts:
-            pretty_print_artifacts_dir()
-        case Choices.training_config:
-            pretty_print_training_config(get_taskgraph_parameters())
-        case Choices.graph:
-            serve_taskgraph_file(
-                parsed_args.graph_url,
-                parsed_args.open_graph,
-                parsed_args.persist_graph,
-                open_in_browser,
-            )
-        case None:
-            pretty_print_task_graph()
-            pretty_print_artifacts_dir()
-            pretty_print_training_config(parameters)
-            serve_taskgraph_file(
-                parsed_args.graph_url,
-                parsed_args.open_graph,
-                parsed_args.persist_graph,
-                open_in_browser,
-            )
+    choice = Choices[parsed_args.only] if parsed_args.only else None
+    if choice == Choices.task_group:
+        pretty_print_task_graph()
+    elif choice == Choices.artifacts:
+        pretty_print_artifacts_dir()
+    elif choice == Choices.training_config:
+        pretty_print_training_config(get_taskgraph_parameters())
+    elif choice == Choices.graph:
+        serve_taskgraph_file(
+            parsed_args.graph_url,
+            parsed_args.open_graph,
+            parsed_args.persist_graph,
+            open_in_browser,
+        )
+    elif choice is None:
+        pretty_print_task_graph()
+        pretty_print_artifacts_dir()
+        pretty_print_training_config(parameters)
+        serve_taskgraph_file(
+            parsed_args.graph_url,
+            parsed_args.open_graph,
+            parsed_args.persist_graph,
+            open_in_browser,
+        )
 
 
 if __name__ == "__main__":