Fix bug in sp500 change detection.

francisco3511 · Feb 20, 2025 · 4b6c821 · 4b6c821
1 parent 1c705bc
commit 4b6c821
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -16,7 +16,7 @@ dependencies = [
     "loguru",
     "tqdm",
     "streamlit",
-    "yfinance",
+    "yfinance >=0.2.54",
     "beautifulsoup4",
     "requests_cache",
     "requests_ratelimiter",

diff --git a/stocksense/pipeline/etl.py b/stocksense/pipeline/etl.py
@@ -94,10 +94,9 @@ def _delist_stocks(
         for tic in last_constituents:
             if tic not in current_constituents:
                 if tic in removals_list:
-                    removed_date = dt.datetime.strptime(
-                        removals.filter(pl.col("name_removed") == tic)["removed"].to_list()[0],
-                        "%Y-%m-%d",
-                    ).date()
+                    removed_date = removals.filter(pl.col("tic") == tic)["removed"].to_list()[0]
+                    if isinstance(removed_date, str):
+                        removed_date = dt.datetime.strptime(removed_date, "%Y-%m-%d").date()
                     self.db.update_stock(tic, {"date_removed": removed_date})
                 else:
                     today = dt.datetime.now().date()