Fix unit test for test_no_mdd_failures

Author: rlarkdfus

vali_objects/utils/live_price_fetcher.py

@@ -88,8 +88,6 @@ def get_latest_price(self, trade_pair: TradePair, time_ms=None) -> Tuple[float,
         Gets the latest price for a single trade pair by utilizing WebSocket and possibly REST data sources.
         Tries to get the price as close to time_ms as possible.
         """
-        # if not time_ms:
-        #     time_ms = TimeUtil.now_in_millis()
         price_sources = self.get_sorted_price_sources_for_trade_pair(trade_pair, time_ms)
         winning_event = PriceSource.get_winning_event(price_sources, time_ms)
         return winning_event.parse_best_best_price_legacy(time_ms), price_sources

vali_objects/vali_dataclasses/price_source.py

@@ -4,6 +4,7 @@
 import bittensor as bt
 from typing import Optional
 from pydantic import BaseModel
+from time_util.time_util import TimeUtil
 
 from vali_objects.enums.order_type_enum import OrderType
 
@@ -55,14 +56,18 @@ def end_ms(self):
     def get_start_time_ms(self):
         return self.start_ms
 
-    def time_delta_from_now_ms(self, now_ms: int) -> int:
+    def time_delta_from_now_ms(self, now_ms:int = None) -> int:
+        if not now_ms:
+            now_ms = TimeUtil.now_in_millis()
         if self.websocket:
             return abs(now_ms - self.start_ms)
         else:
             return min(abs(now_ms - self.start_ms),
                        abs(now_ms - self.end_ms))
 
     def parse_best_best_price_legacy(self, now_ms: int):
+        if not now_ms:
+            now_ms = TimeUtil.now_in_millis()
         if self.websocket:
             return self.open
         else:

vali_objects/vali_dataclasses/recent_event_tracker.py

@@ -80,7 +80,11 @@ def get_closest_event(self, timestamp_ms):
         #print(f"Looking for event at {TimeUtil.millis_to_formatted_date_str(timestamp_ms)}")
         if self.count_events() == 0:
             return None
+
         # Find the event closest to the given timestamp
+        if not timestamp_ms:
+            timestamp_ms = TimeUtil.now_in_millis()
+
         idx = self.events.bisect_left((timestamp_ms,))
         if idx == 0:
             return self.events[0][1]