Télécharger un artifact depuis un registre

Utilisez le SDK Python W&B pour télécharger un artifact lié à un registre. Pour télécharger et utiliser un artifact, vous devez connaître le nom du registre, le nom de la collection, ainsi que l’alias ou l’index de la version de l’artifact à télécharger. Une fois ces propriétés de l’artifact connues, vous pouvez construire le chemin vers l’artifact lié et télécharger l’artifact. Vous pouvez également copier-coller un extrait de code prégénéré depuis l’interface de l’application W&B pour télécharger un artifact lié à un registre.

Construire le chemin d’un artifact lié

Pour télécharger un artifact lié à un registre, vous devez connaître le chemin de cet artifact lié. Ce chemin se compose du nom du registre, du nom de la collection, ainsi que de l’alias ou de l’index de la version de l’artifact à laquelle vous souhaitez accéder. Une fois que vous connaissez le registre, la collection et l’alias ou l’index de la version de l’artifact, vous pouvez construire le chemin de l’artifact lié à l’aide du modèle de chaîne suivant :

# Nom de l’artifact avec indication de l’index de version
f"wandb-registry-{REGISTRY}/{COLLECTION}:v{INDEX}"

# Nom de l’artifact avec indication de l’alias
f"wandb-registry-{REGISTRY}/{COLLECTION}:{ALIAS}"

Remplacez les valeurs entre accolades {} par le nom du registre et de la collection, ainsi que par l’alias ou l’index de la version de l’artifact à laquelle vous souhaitez accéder. Une fois le chemin de l’artifact lié obtenu, utilisez la méthode wandb.Run.use_artifact() pour accéder à l’artifact et télécharger son contenu. L’extrait de code suivant montre comment utiliser et télécharger un artifact lié au registre. Veillez à remplacer les valeurs entre <> par les vôtres :

import wandb

REGISTRY = '<registry_name>'
COLLECTION = '<collection_name>'
ALIAS = '<artifact_alias>'

with wandb.init(entity = '<team_name>', project = '<project_name>')  as run:
    artifact_name = f"wandb-registry-{REGISTRY}/{COLLECTION}:{ALIAS}"
    # artifact_name = '<artifact_name>' # Copiez-collez le nom complet (Full name) indiqué dans l’interface de Registry
    fetched_artifact = run.use_artifact(artifact_or_name = artifact_name)  
    download_path = fetched_artifact.download()  

La méthode wandb.Run.use_artifact() crée un run et marque également l’artifact que vous téléchargez comme entrée de ce run. En marquant un artifact comme entrée d’un run, vous permettez à W&B d’assurer la traçabilité de cet artifact. Si vous ne souhaitez pas créer de run, vous pouvez utiliser l’objet wandb.Api() pour accéder à l’artifact :

import wandb

REGISTRY = "<registry_name>"
COLLECTION = "<collection_name>"
VERSION = "<version>"

api = wandb.Api()
artifact_name = f"wandb-registry-{REGISTRY}/{COLLECTION}:{VERSION}"
artifact = api.artifact(name = artifact_name)

Consultez wandb.Run.use_artifact() et Artifact.download() dans la référence de l’API pour connaître les paramètres et le type de retour.

Utilisateurs disposant d’une entity personnelle et membres de plusieurs organisationsLes utilisateurs disposant d’une entity personnelle et membres de plusieurs organisations doivent également indiquer le nom de leur organisation ou utiliser une entity d’équipe pour accéder aux artifacts liés à un registre.

import wandb

REGISTRY = "<registry_name>"
COLLECTION = "<collection_name>"
VERSION = "<version>"

# Assurez-vous d’utiliser votre entity d’équipe pour instancier l’API
api = wandb.Api(overrides={"entity": "<team-entity>"})
artifact_name = f"wandb-registry-{REGISTRY}/{COLLECTION}:{VERSION}"
artifact = api.artifact(name = artifact_name)

# Construire le chemin
api = wandb.Api()
artifact_name = f"wandb-registry-{REGISTRY}/{COLLECTION}:{VERSION}"
artifact = api.artifact(name = artifact_name)

Copier-coller un extrait de code prégénéré

W&B génère un extrait de code que vous pouvez copier-coller dans votre script Python, votre notebook ou votre terminal pour télécharger un artifact lié à un registre.

Accédez à Registry à l’adresse https://forge.coreweave.com/registry/.
Sélectionnez le nom du registre qui contient votre artifact.
Sélectionnez le nom de la collection.
Dans la liste des versions d’artifact, sélectionnez la version à laquelle vous souhaitez accéder.
Sélectionnez l’onglet Usage.
Copiez l’extrait de code affiché dans la section Usage API.
Collez l’extrait de code dans votre script Python, votre notebook ou votre terminal.

​Construire le chemin d’un artifact lié

​Copier-coller un extrait de code prégénéré

Construire le chemin d’un artifact lié

Copier-coller un extrait de code prégénéré