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 10 wallet holding records for a list of wallet addresses that were active at T_cutoff.
+        Fetches top 2 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.
         """
@@ -360,7 +360,7 @@ class DataFetcher:
             # --- 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 10 per wallet.
+            # 3. Rank these active holdings by USD volume and take the top 2 per wallet.
             query = """
             WITH point_in_time_holdings AS (
                 SELECT
@@ -380,7 +380,7 @@ class DataFetcher:
             )
             SELECT *
             FROM ranked_active_holdings
-            WHERE rn_per_wallet <= 10;
+            WHERE rn_per_wallet <= 2;
             """
             params = {'addresses': batch_addresses, 'T_cutoff': T_cutoff}