Honor YOLO preflight environment config

2026-07-09 02:28:29 +02:00
parent 20c942df50
commit f9493479c7
5 changed files with 45 additions and 6 deletions
@@ -19,7 +19,7 @@ def main() -> int:
    parser.add_argument("--model-path", help="Existing local YOLO model path.")
    parser.add_argument("--tile-manifest-path", help="Existing raster tile manifest path.")
    parser.add_argument("--enabled", action="store_true", help="Treat YOLO as enabled for this preflight.")
-    parser.add_argument("--max-tiles", type=int, default=100, help="Maximum tile count allowed by preflight.")
+    parser.add_argument("--max-tiles", type=int, help="Maximum tile count allowed by preflight.")
    parser.add_argument(
        "--assume-dependencies",
        action="store_true",
@@ -35,11 +35,16 @@ def main() -> int:
    if args.check_model_load and args.assume_dependencies:
        parser.error("--check-model-load cannot be combined with --assume-dependencies")

-    settings = Settings(
-        yolo_enabled=args.enabled or bool(args.model_path),
-        yolo_model_path=args.model_path,
-        yolo_max_tiles=args.max_tiles,
-    )
+    settings = Settings()
+    settings_updates = {}
+    if args.enabled or args.model_path:
+        settings_updates["yolo_enabled"] = True
+    if args.model_path:
+        settings_updates["yolo_model_path"] = args.model_path
+    if args.max_tiles is not None:
+        settings_updates["yolo_max_tiles"] = args.max_tiles
+    if settings_updates:
+        settings = settings.model_copy(update=settings_updates)
    payload = YoloPreflightService.run(
        settings=settings,
        tile_manifest_path=args.tile_manifest_path,