Skip to content

Commit 3f4705c

Browse files
committed
remove the 0th step to match summarize_metrics.py
1 parent 5ecef8b commit 3f4705c

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

.github/workflows/baselines/test_maxtext_metrics.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@ def test_loss(baseline_filename):
3131
loss_expected = {step: loss_expected_values[i] for i, step in enumerate(
3232
range(start_step, end_step+1, interval))}
3333
loss_actual = test_utils.read_maxtext_tb_tag(event_file, loss_summary_name)
34+
del loss_actual[0] # removing the very first step
3435
assert loss_expected.keys() == loss_actual.keys(), \
3536
f"Steps at which loss was emitted for run do not match baseline. \
3637
Actual steps: {loss_actual.keys()}, Baseline steps: {loss_expected.keys()}"

0 commit comments

Comments
 (0)