Fonctions distantes

Ce tutoriel explique comment exécuter des fonctions Python dans une sandbox à l’aide de fonctions distantes. Au lieu de construire des chaînes de commandes shell, vous pouvez appeler le code exécuté dans la sandbox comme s’il s’agissait d’une fonction locale. Il s’adresse aux développeurs qui souhaitent solliciter des ressources de calcul dans une sandbox directement depuis leur code Python et récupérer des résultats typés.

Choisir l’authentification

Ces exemples s’exécutent sur une capacité sans serveur, sauf si une section indique CKS. Choisissez l’identifiant d’authentification du compte propriétaire de vos bacs à sable. Pour obtenir une clé ou un jeton, consultez Choisir et obtenir des identifiants d’authentification. Exécutez les commandes d’installation dans un environnement virtuel Python actif. Enregistrez le code Python de l’onglet correspondant à votre identifiant d’authentification sous le nom sandbox_auth.py, dans le même répertoire que les scripts Python que vous exécutez à partir de cette page. Ce fichier enregistre la méthode d’authentification choisie, et non votre identifiant d’authentification lui-même. Les exemples importent AUTH depuis ce fichier et le transmettent explicitement à chaque nouvelle sandbox ou session.

Clé API Forge
Jeton d’accès à l’API CoreWeave

Utilisez votre clé API Forge avec la stratégie d’authentification W&B du SDK :

uv pip install 'cwsandbox[wandb]>=1.14.2'
export WANDB_API_KEY="[WANDB-API-KEY]"

sandbox_auth.py

from cwsandbox import AuthStrategy

AUTH = AuthStrategy.WANDB

Utilisez le jeton de votre organisation CoreWeave :

uv pip install 'cwsandbox>=1.14.2'
export CWSANDBOX_API_KEY="[API-ACCESS-TOKEN]"

sandbox_auth.py

from cwsandbox import AuthStrategy

AUTH = AuthStrategy.COREWEAVE_API_KEY

Pour un placement sur CKS, choisissez le jeton CoreWeave, activez un exécuteur et définissez placement_mode="cks" lors de la création de la sandbox ou dans SandboxDefaults. Une session créée avec auth=AUTH authentifie également les bacs à sable qui lui appartiennent. Utilisez le même identifiant d’authentification pour vous reconnecter à des bacs à sable, les lister ou les nettoyer.

Définir et appeler une fonction distante

Décorez une fonction avec @session.function() pour l’appeler dans la sandbox avec .remote(). L’appel renvoie une référence, que vous résolvez ensuite avec .result() :

from sandbox_auth import AUTH
from cwsandbox import Sandbox, SandboxDefaults

with Sandbox.session(SandboxDefaults(
    resources={"cpu": "1", "memory": "1Gi"},
), auth=AUTH) as session:
    @session.function()
    def add(x: int, y: int) -> int:
        return x + y

    result = add.remote(2, 3).result()  # 5

Exécuter des fonctions en parallèle avec `.map()`

Utilisez .map() pour lancer de nombreux appels à la fois et récupérer leurs résultats. Les appels indépendants s’exécutent simultanément plutôt que les uns après les autres :

@session.function()
def square(x: int) -> int:
    return x * x

refs = square.map([(1,), (2,), (3,)])
results = [r.result() for r in refs]  # [1, 4, 9]

Étapes suivantes

Vous pouvez désormais appeler des fonctions Python dans une sandbox comme s’il s’agissait d’appels locaux, et paralléliser le travail avec .map(). Pour en savoir plus sur les modes de sérialisation, les tests avec .local() et d’autres sujets, consultez le guide des fonctions distantes.

Sandbox

Utiliser les Sandbox

Gérer les bacs à sable

Référence

Choisir l’authentification

Définir et appeler une fonction distante

Exécuter des fonctions en parallèle avec `.map()`

Étapes suivantes

​Choisir l’authentification

​Définir et appeler une fonction distante

​Exécuter des fonctions en parallèle avec .map()

​Étapes suivantes

Choisir l’authentification

Définir et appeler une fonction distante

Exécuter des fonctions en parallèle avec `.map()`

Étapes suivantes