buildai-cli 0.3.65__tar.gz → 0.3.67__tar.gz

{buildai_cli-0.3.65 → buildai_cli-0.3.67}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: buildai-cli
-Version: 0.3.65
+Version: 0.3.67
 Summary: Build AI CLI (Typer)
 Requires-Python: >=3.11
 Requires-Dist: httpx>=0.27.0

{buildai_cli-0.3.65 → buildai_cli-0.3.67}/cli/commands/processing.py

@@ -194,7 +194,7 @@ def queue_reprioritize(
 @scheduler_app.command("tick")
 def scheduler_tick(
     pipeline_key: str = typer.Option("egoexo", "--pipeline-key", help="Pipeline key."),
-    run_key: str = typer.Option("preview-p1", "--run-key", help="Pipeline run key."),
+    run_key: str | None = typer.Option(None, "--run-key", help="Pipeline run key."),
     stage_key: str | None = typer.Option(None, "--stage-key", help="Single stage to tick."),
     stages: str | None = typer.Option(
         None,
@@ -292,7 +292,7 @@ def scheduler_tick(
 @stage_jobs_app.command("launch-egoexo-prepare-batch")
 def launch_egoexo_prepare_batch(
     limit: int = typer.Option(10, "--limit", min=1, max=200),
-    run_key: str = typer.Option("aligned-v1", "--run-key", help="Prepare run key to drain."),
+    run_key: str = typer.Option(..., "--run-key", help="Prepare run key to drain."),
     execution_profile_key: str = typer.Option(
         "gcp.us.media_l4",
         "--execution-profile-key",
@@ -328,7 +328,7 @@ def launch_egoexo_prepare_batch(
 @stage_jobs_app.command("launch-egoexo-pose-2d-batch")
 def launch_egoexo_pose_2d_batch(
     limit: int = typer.Option(10, "--limit", min=1, max=200),
-    run_key: str = typer.Option("aligned-v1", "--run-key", help="Pose2D run key to drain."),
+    run_key: str = typer.Option(..., "--run-key", help="Pose2D run key to drain."),
     task_count: int = typer.Option(
         8,
         "--task-count",
@@ -346,6 +346,18 @@ def launch_egoexo_pose_2d_batch(
         "--provisioning-model",
         help="Cloud Batch provisioning model: standard or spot.",
     ),
+    max_active_jobs: int | None = typer.Option(
+        None,
+        "--max-active-jobs",
+        min=1,
+        max=1000,
+        help="Maximum active episode Batch jobs for this execution profile.",
+    ),
+    store_heatmaps: bool = typer.Option(
+        False,
+        "--store-heatmaps",
+        help="Persist pre-decoded Sapiens heatmap logits as optional artifacts.",
+    ),
     lease_seconds: int = typer.Option(24 * 60 * 60, "--lease-seconds", min=600),
     dry_run: bool = typer.Option(False, "--dry-run", help="Preview ready work without leasing."),
     write: bool = typer.Option(False, "--write", help="Launch jobs. Omit for dry-run."),
@@ -363,6 +375,8 @@ def launch_egoexo_pose_2d_batch(
             "task_count": task_count,
             "execution_profile_key": execution_profile_key,
             "provisioning_model": provisioning_model,
+            "max_active_jobs": max_active_jobs,
+            "config": {"store_heatmaps": store_heatmaps},
             "lease_seconds": lease_seconds,
             "dry_run": dry_run or not write,
         },

{buildai_cli-0.3.65 → buildai_cli-0.3.67}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "buildai-cli"
-version = "0.3.65"
+version = "0.3.67"
 description = "Build AI CLI (Typer)"
 requires-python = ">=3.11"
 dependencies = [