Skip to content

Commit 83a1ab8

Browse files
committed
relax max_global_steps
1 parent 3daf9d0 commit 83a1ab8

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/checkpoint_engine/test_special_server_adapter.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -132,13 +132,13 @@ async def test_server_manager_with_resume(
132132

133133
# 3. wait for rollout generate responses finished
134134
outputs = await asyncio.gather(*tasks)
135-
expected_min_steps, expected_max_steps = initial_steps - 1, initial_steps + train_steps - 1
135+
expected_min_steps = initial_steps - 1
136136
for output in outputs:
137137
assert output.min_global_steps == expected_min_steps, (
138138
f"output.min_global_steps is {output.min_global_steps}, expected {expected_min_steps}"
139139
)
140-
assert output.max_global_steps == expected_max_steps, (
141-
f"output.max_global_steps is {output.max_global_steps}, expected {expected_max_steps}"
140+
assert output.max_global_steps > expected_min_steps, (
141+
f"output.max_global_steps is {output.max_global_steps}, expected > {expected_min_steps}"
142142
)
143143
assert output.stop_reason not in ("aborted", "abort"), (
144144
f"output.stop_reason is {output.stop_reason}, expected not abort"

0 commit comments

Comments
 (0)