mjlab/scripts/cloud/sweep-agent.yaml
Upstream Snapshot 32a241c28f
Some checks failed
nightly / Test against latest dependencies (py3.10) (push) Has been cancelled
nightly / Test against latest dependencies (py3.13) (push) Has been cancelled
tests / tests (3.13, locked) (push) Has been cancelled
tests / tests (3.13, unlocked) (push) Has been cancelled
tests / pyright (3.10) (push) Has been cancelled
tests / lint-format (push) Has been cancelled
tests / tests (3.10, locked) (push) Has been cancelled
tests / tests (3.11, locked) (push) Has been cancelled
tests / tests (3.12, locked) (push) Has been cancelled
tests / pyright (3.11) (push) Has been cancelled
tests / pyright (3.12) (push) Has been cancelled
tests / pyright (3.13) (push) Has been cancelled
tests / ty-check (3.10) (push) Has been cancelled
tests / ty-check (3.11) (push) Has been cancelled
tests / ty-check (3.12) (push) Has been cancelled
tests / ty-check (3.13) (push) Has been cancelled
tests / stubs (push) Has been cancelled
tests / smoke-test (push) Has been cancelled
Docker / check_paths (push) Has been cancelled
docs / build (push) Has been cancelled
Docker / build (push) Has been cancelled
Import upstream snapshot c19f713c415a699a79d71cd96aa13c3104a05047
Upstream: https://github.com/michaelgillett/mjlab
Upstream-Commit: c19f713c415a699a79d71cd96aa13c3104a05047
Upstream-Branch: main
2026-08-28 15:42:17 +08:00

18 lines
439 B
YAML

# SkyPilot job that runs a single W&B sweep agent on one GPU.
#
# Submit to an existing cluster provisioned by sweep-cluster.yaml:
# sky exec mjlab-sweep scripts/cloud/sweep-agent.yaml \
# --gpus A100:1 --env SWEEP_ID=<entity/project/sweep_id> -d
resources:
accelerators: A100:1
envs:
SWEEP_ID: ""
MUJOCO_GL: egl
run: |
source "$HOME/.local/bin/env" 2>/dev/null || true
cd ~/sky_workdir
uv run wandb agent "$SWEEP_ID"