fix: choose latest added metric if their timestamps are the same

usnistgov · Jan 30, 2025 · e44c2f6 · e44c2f6
1 parent 693317a
commit e44c2f6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/unit/restapi/lib/mock_mlflow.py b/tests/unit/restapi/lib/mock_mlflow.py
@@ -49,7 +49,9 @@ def get_run(self, id: str) -> MockMlflowRun:
             # find the latest metric for each metric name
             if (
                 metric.key not in output_metrics
-                or metric.timestamp > output_metrics[metric.key].timestamp
+                # use >= here since we append to the list in log_metric and want to make sure we 
+                # return the latest metric available if they have the same timestamp
+                or metric.timestamp >= output_metrics[metric.key].timestamp 
             ):
                 output_metrics[metric.key] = metric