add metrics_only flag

Signed-off-by: Andrew Xue <andrewxue@anyscale.com>
ray-project · Oct 16, 2023 · 0e018c6 · 0e018c6
1 parent 70cf98f
commit 0e018c6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/ray/data/_internal/stats.py b/python/ray/data/_internal/stats.py
@@ -284,7 +284,7 @@ def update_stats_actor_metrics(
     metric_keys = OpRuntimeMetrics.get_metric_keys()
     stats = {key: 0 for key in metric_keys}
     for op_metric in op_metrics:
-        metric_dict = op_metric.as_dict()
+        metric_dict = op_metric.as_dict(metrics_only=True)
         for key in metric_keys:
             stats[key] += metric_dict.get(key, 0)