Upload data/data_fetcher.py with huggingface_hub

data/data_fetcher.py

@@ -342,7 +342,7 @@ class DataFetcher:
 
     def fetch_wallet_holdings(self, wallet_addresses: List[str], T_cutoff: datetime.datetime) -> Dict[str, List[Dict[str, Any]]]:
         """
-        Fetches top 3 wallet holding records for a list of wallet addresses that were active at T_cutoff.
+        Fetches top 10 wallet holding records for a list of wallet addresses that were active at T_cutoff.
         Batches queries to avoid "Max query size exceeded" errors.
         Returns a dictionary mapping wallet_address to a LIST of its holding data.
         """
@@ -357,10 +357,10 @@ class DataFetcher:
         for i in range(0, total_wallets, BATCH_SIZE):
             batch_addresses = wallet_addresses[i : i + BATCH_SIZE]
 
-            # --- NEW: Time-aware query based on user's superior logic ---
+            # --- Time-aware query ---
             # 1. For each holding, find the latest state at or before T_cutoff.
             # 2. Filter for holdings where the balance was greater than 0.
-            # 3. Rank these active holdings by USD volume and take the top 3 per wallet.
+            # 3. Rank these active holdings by USD volume and take the top 10 per wallet.
             query = """
             WITH point_in_time_holdings AS (
                 SELECT
@@ -368,7 +368,7 @@ class DataFetcher:
                     COALESCE(history_bought_cost_sol, 0) + COALESCE(history_sold_income_sol, 0) AS total_volume_usd,
                     ROW_NUMBER() OVER(PARTITION BY wallet_address, mint_address ORDER BY updated_at DESC) as rn_per_holding
                 FROM wallet_holdings
-                WHERE 
+                WHERE
                     wallet_address IN %(addresses)s
                     AND updated_at <= %(T_cutoff)s
             ),
@@ -380,7 +380,7 @@ class DataFetcher:
             )
             SELECT *
             FROM ranked_active_holdings
-            WHERE rn_per_wallet <= 3;
+            WHERE rn_per_wallet <= 10;
             """
             params = {'addresses': batch_addresses, 'T_cutoff': T_cutoff}