Fix occasional gaps in cumulative Goodput Monitor.

PiperOrigin-RevId: 787142345

Author: dipannita08

ml_goodput_measurement/src/gcp_metrics.py

@@ -100,7 +100,7 @@ def send_metrics(self, metrics: list[Dict[str, Any]]):
       self.client.create_time_series(
           name=self.project_name, time_series=time_series_list
       )
-      logger.info("Sent %d metrics to GCP Monitoring.", len(metrics))
+      logger.info("Sent %d Goodput metrics to GCM Monitoring.", len(metrics))
 
     except GoogleAPIError as e:
-      logger.error("Failed to send metrics: %s", e)
+      logger.error("Failed to send Goodput metrics to GCM Monitoring: %s", e)

ml_goodput_measurement/src/goodput.py

@@ -1174,7 +1174,10 @@ def _fetch_new_entries(self, query_time: datetime.datetime) -> list[Any]:
         cached_last_entry_info = self._goodput_cache.get_last_entry_info()
         if cached_last_entry_info:
           cached_last_entry_ts, _ = cached_last_entry_info
-          if query_time <= cached_last_entry_ts:
+          if (
+              cached_last_entry_ts is not None
+              and query_time <= cached_last_entry_ts
+          ):
             return []
 
           new_entries, current_last_entry_info = (

ml_goodput_measurement/src/goodput_cache.py

@@ -34,7 +34,8 @@ def update_cached_entries(
   ):
     """Updated the cached entries."""
     self._cached_entries.extend(entries)
-    self.update_last_entry_info(last_entry_info)
+    if last_entry_info and last_entry_info[0] is not None:
+      self.update_last_entry_info(last_entry_info)
     self.update_job_start_time()
     self.update_job_end_time()
     new_step_entries = [entry for entry in entries if _STEP_START_TIME in entry]