W&B 専用クラウドでサンドボックスを使用する

uv pip install 'cwsandbox>=1.14.2'

export FORGE_HOST="[INSTANCE].wandb.io"
export WANDB_API_KEY="[WANDB-API-KEY]"

import os
from cwsandbox import AuthHeaders, Sandbox

forge_auth = AuthHeaders(
    headers={
        "x-wandb-api-key": os.environ["WANDB_API_KEY"],
        "x-wandb-host": os.environ["FORGE_HOST"],
    },
    strategy="wandb_api_key",
)

with Sandbox.run(
    auth=forge_auth,
    resources={"cpu": "1", "memory": "1Gi"},
) as sandbox:
    result = sandbox.exec(["echo", "Hello!"]).result()
    print(result.stdout)