buildai-cli 0.3.64__tar.gz → 0.3.66__tar.gz

{buildai_cli-0.3.64 → buildai_cli-0.3.66}/PKG-INFO

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

{buildai_cli-0.3.64 → buildai_cli-0.3.66}/cli/commands/processing.py

@@ -330,14 +330,14 @@ 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."),
     task_count: int = typer.Option(
-        96,
+        8,
         "--task-count",
         min=1,
         max=1000,
         help="Cloud Batch tasks per episode job.",
     ),
     execution_profile_key: str = typer.Option(
-        "gcp.us.gpu_l4",
+        "gcp.us.gpu_a100",
         "--execution-profile-key",
         help="Cloud Batch execution profile used for each episode job.",
     ),
@@ -346,11 +346,23 @@ 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."),
 ) -> None:
-    """Launch one 96-task Cloud Batch Pose2D job per ready EgoExo episode."""
+    """Launch one episode-owned Cloud Batch Pose2D job per ready EgoExo episode."""
     if provisioning_model not in {"standard", "spot"}:
         error("--provisioning-model must be one of: standard, spot")
         raise typer.Exit(1)
@@ -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.64 → buildai_cli-0.3.66}/pyproject.toml

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