Fix test_gpu_model_runner.py::test_update_states_request_resumed

David Ben-David · David Ben-David · commit 3af28dc2e5c8 · 2025-09-29T18:20:56.000+03:00
Signed-off-by: David Ben-David &lt;davidb@pliops.com&gt;
diff --git a/tests/v1/worker/test_gpu_model_runner.py b/tests/v1/worker/test_gpu_model_runner.py
@@ -251,6 +251,7 @@ def test_update_states_request_resumed(model_runner, dist_init):
         new_token_ids=[[]],
         new_block_ids=([[0]], ),
         num_computed_tokens=[0],
+        num_output_tokens=[0],
     )
 
     scheduler_output = SchedulerOutput(

Original file line number	Diff line number	Diff line change
`@@ -251,6 +251,7 @@ def test_update_states_request_resumed(model_runner, dist_init):`
`251`	`251`	`new_token_ids=[[]],`
`252`	`252`	`new_block_ids=([[0]], ),`
`253`	`253`	`num_computed_tokens=[0],`
	`254`	`+ num_output_tokens=[0],`
`254`	`255`	`)`
`255`	`256`
`256`	`257`	`scheduler_output = SchedulerOutput(`