enable dotenv to load env vars during training

henryaddison · Oct 28, 2024 · 344f9a9 · 344f9a9
1 parent 276cb37
commit 344f9a9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/bin/main.py b/bin/main.py
@@ -22,9 +22,12 @@
 from ml_collections.config_flags import config_flags
 import logging
 import os
+from dotenv import load_dotenv
 
 from knockknock import slack_sender
 
+load_dotenv()  # take environment variables from .env
+
 FLAGS = flags.FLAGS
 
 config_flags.DEFINE_config_file(