Utiliser les sandbox avec Cloud dédié de W&B

Si votre organisation utilise Cloud dédié de W&B, authentifiez les sandbox exécutés sur une capacité sans serveur à l’aide d’une clé API issue de votre instance dédiée, et non de wandb.ai. Le client Python envoie votre clé et le nom d’hôte de votre instance au service CoreWeave Sandboxes, qui valide la clé auprès de votre instance. Transmettez le nom d’hôte de votre instance avec la clé afin que l’authentification s’effectue auprès de la bonne instance.

Avant de commencer

Vous avez besoin des éléments suivants :

L’accès aux sandboxes activé pour votre instance de Cloud dédié. Contactez votre équipe de compte pour vérifier que votre instance dispose des mises à jour serveur requises et pour demander l’accès.
L’URL de votre instance de Cloud dédié, par exemple https://[INSTANCE].wandb.io. Le nom d’hôte doit se terminer par .wandb.io. Les instances utilisant un domaine personnalisé ne sont pas prises en charge.
Python 3.11 ou version ultérieure.

Les déploiements autogérés de W&B, y compris les installations isolées du réseau, ne sont pas pris en charge. Le client TypeScript ne prend pas encore en charge les instances de Cloud dédié : utilisez donc le client Python.

Obtenir une clé API depuis votre instance

Créez une clé API personnelle sur votre instance dédiée, et non sur forge.coreweave.com/settings#apikeys :

Connectez-vous à votre instance de Cloud dédié.
Cliquez sur le sélecteur de compte dans le coin supérieur droit, puis sur Paramètres utilisateur.
Dans la section Clés API, cliquez sur Nouvelle clé. Nommez la clé, puis cliquez sur Créer une clé API.

Votre instance n’affiche la clé complète qu’une seule fois, au moment de sa création. Pour plus d’informations, voir Comment puis-je trouver ma clé API ? dans la documentation Forge.

Configurer l’authentification de l’instance

Installez le client Python :

uv pip install 'cwsandbox>=1.14.2'

Définissez le nom d’hôte de votre instance et votre clé API. Remplacez [INSTANCE] par le préfixe du nom d’hôte de votre instance et [WANDB-API-KEY] par la clé que vous avez créée :

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

L’exemple transmet le nom d’hôte dans l’en-tête x-wandb-host. Indiquez le nom d’hôte sans https:// ni chemin. Le Sandbox API endpoint ne change pas.

Exécuter une sandbox

Transmettez la clé et le nom d’hôte de l’instance dédiée via AuthHeaders :

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)

Si la commande réussit, Hello! s’affiche dans la sortie. En cas d’échec de l’authentification, vérifiez que FORGE_HOST correspond à l’instance sur laquelle vous avez créé la clé. Une clé issue d’une instance dédiée ne permet pas de s’authentifier auprès de wandb.ai, et inversement, une clé wandb.ai ne permet pas de s’authentifier auprès d’une instance dédiée.

Étapes suivantes

Placement et débordement des sandboxes présente les modes de placement, les identifiants d’authentification et le débordement (spillover).
La documentation du client Python traite du cycle de vie, de l’exécution, des opérations sur les fichiers et d’autres modèles d’utilisation du SDK.

Sandbox