Commit bd9009e4 authored by Zachary Charles's avatar Zachary Charles Committed by tensorflow-copybara
Browse files

Add int conversion when clearing metrics in order to circumvent opaque failure modes.

PiperOrigin-RevId: 411610540
parent 89513c89
......@@ -304,7 +304,7 @@ class CSVMetricsManager(metrics_manager.MetricsManager):
latest_round_num = None
for metric_row in metrics:
metric_row_round_num = metric_row['round_num']
metric_row_round_num = int(metric_row['round_num'])
if metric_row_round_num <= round_num:
reduced_fieldnames = reduced_fieldnames.union(metric_row.keys())
reduced_metrics.append(metric_row)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment