Passer au contenu principal
Avec W&B Weave, vous pouvez tracer les appels de fonction dans votre code et les lier directement aux runs W&B dans lesquels 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 les liens vers les runs associés.

​
Afficher un run W&B dans le tableau Traces

Le code Python suivant montre comment Weave lie les Ops tracées aux runs W&B lorsque vous les exécutez dans un contexte wandb.init(). Ces traces apparaissent dans la Weave UI et sont associées au run correspondant.Pour afficher un run W&B sous forme de trace Weave :
  1. Dans le terminal, installez les dépendances.
pip install wandb weave
  1. Connectez-vous à W&B.
wandb login
  1. Dans le script suivant, remplacez your-team-name/your-project-name par votre entité et votre projet W&B :
import wandb
import weave

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

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

    # Définir une 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 ceci par votre nom d'utilisateur/projet W&B réel
    example_wandb("your-team-name/your-project-name")
  1. Exécutez le script.
python weave_trace_with_wandb.py
  1. Accédez à la W&B App et sélectionnez votre projet.
  2. Dans la barre latérale du projet Weave, cliquez sur Traces. Le tableau Traces affiche des liens vers les runs associés.
Vous pouvez maintenant ouvrir n’importe quelle trace depuis le tableau Traces pour examiner le run W&B lié et le contexte d’expérience associé.