File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff 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"
You can’t perform that action at this time.
0 commit comments