From ae9058c01353dc4c045fa50a307c911f13b4cc9e Mon Sep 17 00:00:00 2001 From: YangWu1227 Date: Tue, 30 Jul 2024 20:23:28 -0700 Subject: [PATCH] refactor: resolve mypy warnings --- main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 28d8ac6..84d5c82 100644 --- a/main.py +++ b/main.py @@ -15,8 +15,8 @@ def main() -> int: if ENV == "dev": max_etfs = 20 elif ENV == "prod": - max_etfs = int(os.getenv('MAX_ETFS')) - ipo_date = datetime.strptime(os.getenv('IPO_DATE'), "%Y-%m-%d") + max_etfs = int(os.getenv('MAX_ETFS', 1)) + ipo_date = datetime.strptime(os.getenv('IPO_DATE', datetime.today().strftime('%Y-%m-%d')), "%Y-%m-%d") etfs_data = query_etf_data(logger=logger, max_etfs=max_etfs, ipo_date=ipo_date) s3_bucket = os.getenv("S3_BUCKET")