From b8f1a0c8c05d6d1617b818c45b4c3285a7675156 Mon Sep 17 00:00:00 2001
From: Akshat Bubna <akshatb42@gmail.com>
Date: Thu, 23 May 2024 19:20:17 +0000
Subject: [PATCH] better fix for wandb secret

---
 src/common.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/common.py b/src/common.py
index 45f2fee..25e05f6 100644
--- a/src/common.py
+++ b/src/common.py
@@ -43,7 +43,8 @@
     APP_NAME,
     secrets=[
         modal.Secret.from_name("huggingface"), 
-        (modal.Secret.from_name("wandb") if ALLOW_WANDB else modal.Secret.from_dict({})),
+        modal.Secret.from_dict({"ALLOW_WANDB": os.environ.get("ALLOW_WANDB", "false")}),
+        *([modal.Secret.from_name("wandb")] if ALLOW_WANDB else []),
     ],
 )