to your stack:

zenml integration install wandb -yThe Weights & Biases Experiment Tracker needs to be configured with the credentials required to connect to the Weights & Biases platform using one of the available authentication methods.

Authentication Methods

You need to configure the following credentials for authentication to the Weights & Biases platform:

api_key: Mandatory API key token of your Weights & Biases account.

project_name: The name of the project where you're sending the new run. If the project is not specified, the run is put in an "Uncategorized" project.

entity: An entity is a username or team name where you're sending runs. This entity must exist before you can send runs there, so make sure to create your account or team in the UI before starting to log runs. If you don't specify an entity, the run will be sent to your default entity, which is usually your username.

This option configures the credentials for the Weights & Biases platform directly as stack component attributes.

This is not recommended for production settings as the credentials won't be stored securely and will be clearly visible in the stack configuration.

# Register the Weights & Biases experiment tracker

zenml experiment-tracker register wandb_experiment_tracker --flavor=wandb \

--entity=<entity> --project_name=<project_name> --api_key=<key>

# Register and set a stack with the new experiment tracker

zenml stack register custom_stack -e wandb_experiment_tracker ... --set

This method requires you to configure a ZenML secret to store the Weights & Biases tracking service credentials securely.

You can create the secret using the zenml secret create command:

zenml secret create wandb_secret \

--entity=<ENTITY> \

--project_name=<PROJECT_NAME>

--api_key=<API_KEY>

Once the secret is created, you can use it to configure the wandb Experiment Tracker:

# Reference the entity, project and api-key in our experiment tracker component

zenml experiment-tracker register wandb_tracker \

--flavor=wandb \