scaffold.wandb.helpers

Attributes

WANDB_KEY
WANDB_SECRET
logger

Functions

wandb_environment_setup(→ None)

Retrieve wandb login secrets secrets and export them as environment variables.

Module Contents

scaffold.wandb.helpers.wandb_environment_setup(username: str) → None

Retrieve wandb login secrets secrets and export them as environment variables. This is a convenience function for our Flyte setup to be called before wandb.init().

If this function is called in local setup, i.e. outside of flyte task context, it will not do anything.

Parameters:

username (str) – WandB user handle for look up of API key

scaffold.wandb.helpers.WANDB_KEY = 'key_map'

scaffold.wandb.helpers.WANDB_SECRET = 'wandb-access-secret'

scaffold.wandb.helpers.logger