Fix download_schwab_token() to return after token write

punitarani · Jul 28, 2024 · e319933 · e319933
1 parent b6bd95c
commit e319933
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/server/load.py b/server/load.py
@@ -11,11 +11,13 @@
 
 def download_schwab_token() -> None:
     schwab_token = os.getenv("SCHWAB_TOKEN", None)
+    if schwab_token is None:
+        raise ValueError("SCHWAB_TOKEN environment variable not set")
+
     if schwab_token is not None:
         with open(SCHWAB_TOKEN_FP, "w") as f:
             f.write(schwab_token)
-
-    raise ValueError("SCHWAB_TOKEN environment variable not set")
+            return
 
 
 async def load_models():