move parser to function

UWARG · Mar 25, 2024 · e8e1246 · e8e1246
1 parent 56a10ac
commit e8e1246
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/blue_only.py b/blue_only.py
@@ -32,15 +32,6 @@
 
 CONFIG_FILE_PATH = pathlib.Path("config.yml")
 
-parser = argparse.ArgumentParser(description="AEAC 2024 Auto-landing script")
-parser.add_argument(
-    "--method",
-    help="Method for selecting landing pads",
-    choices=["contour", "yolo"],
-    default="contour",
-)
-args = parser.parse_args()
-
 
 def current_milli_time() -> int:
     """
@@ -367,6 +358,15 @@ def main() -> int:
     except IOError as exc:
         print(f"Error when opening file: {exc}")
 
+    parser = argparse.ArgumentParser(description="AEAC 2024 Auto-landing script")
+    parser.add_argument(
+        "--method",
+        help="Method for selecting landing pads",
+        choices=["contour", "yolo"],
+        default="contour",
+    )
+    args = parser.parse_args()
+
     if args.method == "contour":
         return run_with_contours()