buildai-cli 0.3.152__tar.gz → 0.3.154__tar.gz

{buildai_cli-0.3.152 → buildai_cli-0.3.154}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: buildai-cli
-Version: 0.3.152
+Version: 0.3.154
 Summary: Build AI CLI (Typer)
 Requires-Python: >=3.11
 Requires-Dist: google-crc32c>=1.7.1

{buildai_cli-0.3.152 → buildai_cli-0.3.154}/cli/commands/egoexo_config.py

@@ -333,12 +333,14 @@ def set_recipe(
 
 @app.command("set-task-classes")
 def set_task_classes(
-    file: Path = typer.Argument(..., help="Task class list JSON, one to three classes."),
+    file: Path = typer.Argument(
+        ..., help="Per-class overlays keyed by name, for example limits; images stay in env."
+    ),
     uri: str | None = URI_OPTION,
     write: bool = WRITE_OPTION,
     reason: str | None = REASON_OPTION,
 ) -> None:
-    """Store task classes that override BUILDAI_EGOEXO_ASSIGNED_TASK_CLASSES_JSON."""
+    """Store per-class field overlays applied on top of the environment task classes."""
     classes = _load_json_file(file)
     _revise(
         _store(uri),

{buildai_cli-0.3.152 → buildai_cli-0.3.154}/cli/commands/processing.py

@@ -16,10 +16,8 @@ from cli.console import error
 app = typer.Typer(help="Inspect and adjust processing manifests.", no_args_is_help=True)
 queue_app = typer.Typer(help="Inspect and adjust manifest batch queues.", no_args_is_help=True)
 scheduler_app = typer.Typer(help="Run bounded processing scheduler passes.", no_args_is_help=True)
-stage_jobs_app = typer.Typer(help="Operate episode-scoped stage jobs.", no_args_is_help=True)
 app.add_typer(queue_app, name="queue")
 app.add_typer(scheduler_app, name="scheduler")
-app.add_typer(stage_jobs_app, name="stage-jobs")
 
 
 def _request(
@@ -287,153 +285,3 @@ def scheduler_tick(
             ),
         },
     )
-
-
-@stage_jobs_app.command("tick-egoexo")
-def tick_egoexo_stage_jobs(
-    run_key: str | None = typer.Option(None, "--run-key", help="Stage-job run key to tick."),
-    program_slug: str = typer.Option(
-        "build-egoexo-2026",
-        "--program-slug",
-        help="Program slug whose active episodes should be planned.",
-    ),
-    recording_batch_id: str | None = typer.Option(
-        None,
-        "--recording-batch-id",
-        help="Optional recording batch id to plan exactly one finalized ingest batch.",
-    ),
-    stages: str | None = typer.Option(
-        None,
-        "--stages",
-        help="Comma-separated canonical stage keys. Defaults to the full recording-scoped pipeline.",
-    ),
-    plan_limit: int = typer.Option(500, "--plan-limit", min=1, max=5000),
-    priority: int = typer.Option(7, "--priority", min=0, max=10),
-    skip_existing_pose_3d: bool = typer.Option(
-        True,
-        "--skip-existing-pose-3d/--include-existing-pose-3d",
-        help="Skip Pose3D work for episodes that already have Pose3D outputs.",
-    ),
-    launch: bool = typer.Option(
-        False, "--launch/--no-launch", help="Launch ready workers after planning."
-    ),
-    dry_run: bool = typer.Option(
-        False, "--dry-run", help="Preview the tick without writes or launches."
-    ),
-    write: bool = typer.Option(False, "--write", help="Create work items. Omit for dry-run."),
-) -> None:
-    """Run the canonical recording-scoped EgoExo stage-job tick."""
-    _request(
-        "/v1/processing/stage-jobs/egoexo/tick",
-        method="POST",
-        body={
-            "program_slug": program_slug,
-            "recording_batch_id": recording_batch_id,
-            "stages": _parse_csv(
-                stages,
-                option_name="--stages",
-                allowed={"frame_extract", "pose_2d", "pose_3d_definition_run"},
-            ),
-            "run_key": run_key,
-            "plan_limit": plan_limit,
-            "priority": priority,
-            "skip_existing_pose_3d": skip_existing_pose_3d,
-            "launch_frame_extract": launch,
-            "launch_pose_2d": launch,
-            "launch_pose_3d": launch,
-            "dry_run": dry_run or not write,
-        },
-    )
-
-
-@stage_jobs_app.command("launch-egoexo-pose-2d-modal")
-def launch_egoexo_pose_2d_modal(
-    limit: int = typer.Option(500, "--limit", min=1, max=1000),
-    run_key: str = typer.Option(..., "--run-key", help="Pose2D run key to drain."),
-    task_count: int = typer.Option(
-        500,
-        "--task-count",
-        min=1,
-        max=1000,
-        help="Maximum Modal workers per episode work item.",
-    ),
-    execution_profile_key: str = typer.Option(
-        "modal.h100.500",
-        "--execution-profile-key",
-        help="Modal execution profile used for each episode job.",
-    ),
-    store_heatmaps: bool = typer.Option(
-        False,
-        "--store-heatmaps",
-        help="Persist pre-decoded Sapiens heatmap logits as optional artifacts.",
-    ),
-    modal_spawn_parallelism: int | None = typer.Option(
-        None,
-        "--modal-spawn-parallelism",
-        min=1,
-        max=1000,
-        help="Maximum concurrent Modal spawn calls for this launcher request.",
-    ),
-    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 Modal-backed Sapiens Pose2D workers for ready EgoExo episodes."""
-    config: dict[str, object] = {"store_heatmaps": store_heatmaps}
-    if modal_spawn_parallelism is not None:
-        config["modal_spawn_parallelism"] = modal_spawn_parallelism
-    _request(
-        "/v1/processing/stage-jobs/egoexo/pose-2d/launch-modal",
-        method="POST",
-        body={
-            "limit": limit,
-            "run_key": run_key,
-            "task_count": task_count,
-            "execution_profile_key": execution_profile_key,
-            "config": config,
-            "lease_seconds": lease_seconds,
-            "dry_run": dry_run or not write,
-        },
-    )
-
-
-@stage_jobs_app.command("launch-egoexo-pose-3d-batch")
-def launch_egoexo_pose_3d_batch(
-    limit: int = typer.Option(10, "--limit", min=1, max=200),
-    run_key: str = typer.Option(..., "--run-key", help="Pose3D run key to drain."),
-    execution_profile_key: str = typer.Option(
-        "gcp.us.pose_3d_cpu",
-        "--execution-profile-key",
-        help="Cloud Batch execution profile used for each episode triangulation job.",
-    ),
-    provisioning_model: str = typer.Option(
-        "standard",
-        "--provisioning-model",
-        help="Cloud Batch provisioning model: standard or spot.",
-    ),
-    config_json: str | None = typer.Option(
-        None,
-        "--config-json",
-        help="Optional Pose3D processor config JSON object.",
-    ),
-    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 optimized receipt-backed Pose3D triangulation jobs for ready episodes."""
-    if provisioning_model not in {"standard", "spot"}:
-        error("--provisioning-model must be one of: standard, spot")
-        raise typer.Exit(1)
-    _request(
-        "/v1/processing/stage-jobs/egoexo/pose-3d/launch-batch",
-        method="POST",
-        body={
-            "limit": limit,
-            "run_key": run_key,
-            "execution_profile_key": execution_profile_key,
-            "provisioning_model": provisioning_model,
-            "config": _parse_json_object(config_json, option_name="--config-json"),
-            "lease_seconds": lease_seconds,
-            "dry_run": dry_run or not write,
-        },
-    )

{buildai_cli-0.3.152 → buildai_cli-0.3.154}/pyproject.toml

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