Lier un run W&B pour tracer les appels de fonctions

Avec W&B Weave, vous pouvez tracer les appels de fonctions dans votre code et les lier directement aux runs W&B au cours desquels ils ont été exécutés. Lorsque vous tracez une fonction avec @weave.op() et que vous l’appelez dans un contexte wandb.init(), Weave associe automatiquement la trace au run W&B. Le tableau Traces affiche des liens vers les runs associés.

Afficher un run W&B dans le tableau Traces

Python
TypeScript

Le code Python suivant montre comment Weave lie les ops tracés aux runs W&B lorsque vous les exécutez dans un contexte wandb.init(). Ces traces apparaissent dans l’interface de Weights & Biases et sont associées au run correspondant.Pour afficher un run W&B sous forme de trace Weave :

Dans le terminal, installez les dépendances.

pip install wandb weave

Connectez-vous à Forge.

wandb login

Dans le script suivant, remplacez your-team-name/your-project-name par votre entity et votre projet W&B :

import wandb
import weave

def example_wandb(projname):
    # Séparer projname en entity et projet
    entity, project = projname.split("/", 1)

    # Initialiser le contexte Weave pour le traçage
    weave.init(projname)

    # Définir un op traçable
    @weave.op()
    def say(message: str) -> str:
        return f"I said: {message}"

    # Premier run W&B
    with wandb.init(
        entity=entity,
        project=project,
        notes="Experiment 1",
        tags=["baseline", "paper1"],
    ) as run:
        say("Hello, world!")
        say("How are you!")
        run.log({"messages": 2})

    # Deuxième run W&B
    with wandb.init(
        entity=entity,
        project=project,
        notes="Experiment 2",
        tags=["baseline", "paper1"],
    ) as run:
        say("Hello, world from experiment 2!")
        say("How are you!")
        run.log({"messages": 2})

if __name__ == "__main__":
    # Remplacez cette valeur par votre nom d’utilisateur/projet W&B réel
    example_wandb("your-team-name/your-project-name")

Exécutez le script.

python weave_trace_with_wandb.py

Accédez à la W&B App et sélectionnez votre projet.
Dans la barre latérale du projet Weave, cliquez sur Traces. Le tableau Traces affiche des liens vers tous les runs associés.

Vous pouvez désormais ouvrir n’importe quelle trace depuis le tableau Traces pour inspecter le run W&B lié et examiner le contexte d’expérience associé.

This feature is not available for the TypeScript SDK yet.

​Afficher un run W&B dans le tableau Traces

Afficher un run W&B dans le tableau Traces