better way to handle FutureWarnings. fix expected input type

bug-or-feature · Jan 21, 2025 · 9685146 · 9685146
1 parent f04723d
commit 9685146
Showing 1 changed file with 9 additions and 10 deletions.
diff --git a/sysproduction/update_historical_prices.py b/sysproduction/update_historical_prices.py
@@ -482,20 +482,19 @@ def write_merged_prices_for_contract(
     diag_prices = diagPrices(data)
     price_updater = updatePrices(data)
 
-    list_of_data = []
-    for frequency in list_of_frequencies:
-        prices = diag_prices.get_prices_at_frequency_for_contract_object(
+    list_of_data = [
+        diag_prices.get_prices_at_frequency_for_contract_object(
             contract_object,
             frequency=frequency,
         )
-        if len(prices) > 0:
-            list_of_data.append(prices)
+        for frequency in list_of_frequencies
+    ]
 
-    if list_of_data:
-        merged_prices = merge_data_with_different_freq(list_of_data)
-        price_updater.overwrite_merged_prices_for_contract(
-            contract_object=contract_object, new_prices=merged_prices
-        )
+    merged_prices = merge_data_with_different_freq(list_of_data)
+
+    price_updater.overwrite_merged_prices_for_contract(
+        contract_object=contract_object, new_prices=futuresContractPrices(merged_prices)
+    )
 
 
 if __name__ == "__main__":