From f4434f8e3087ed352b757ba076543c92c168cb58 Mon Sep 17 00:00:00 2001
From: Justin Brooks <jsbroks@gmail.com>
Date: Wed, 13 Dec 2023 22:52:38 -0500
Subject: [PATCH] Add extra envs

---
 main.tf      | 2 ++
 variables.tf | 5 +++++
 2 files changed, 7 insertions(+)

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