diff --git a/main.tf b/main.tf index eef80e90..04073c2f 100644 --- a/main.tf +++ b/main.tf @@ -201,6 +201,8 @@ module "wandb" { host = local.url license = var.license + extraEnv = var.other_wandb_env + bucket = { provider = "s3" name = local.bucket_name diff --git a/variables.tf b/variables.tf index 5622cf5d..12404d43 100644 --- a/variables.tf +++ b/variables.tf @@ -332,3 +332,8 @@ variable "license" { description = "Weights & Biases license key." } +variable "other_wandb_env" { + type = map(any) + description = "Extra environment variables for W&B" + default = {} +} \ No newline at end of file