🌲 Bug: Fix Env Var Issue 🌲 (#53)

suchak1 · Oct 28, 2020 · 74e3842 · 74e3842
1 parent 4ae86d8
commit 74e3842
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/DataSource.py b/src/DataSource.py
@@ -390,7 +390,7 @@ def get_intraday(self, symbol, min=1, timeframe='max', extra_hrs=True):
 
 
 class Polygon(MarketData):
-    def __init__(self, token=os.environ['APCA_API_KEY_ID']):
+    def __init__(self, token=os.environ.get('APCA_API_KEY_ID')):
         load_dotenv()
         super().__init__()
         self.client = RESTClient(token)