[fix] Fix test_connector_scheduler_output

- Do not expect an additional token when using the overlap scheduler.

Signed-off-by: Robin Kobus <[email protected]>

Author: Funatiq

tests/integration/defs/llmapi/test_llm_api_connector.py

@@ -238,9 +238,7 @@ def test_connector_scheduler_output(enforce_single_worker, model_with_connector,
         scheduler.update_state_after_alloc.call_args.args[1]) == math.ceil(
             NUM_INPUT_TOKENS / BLOCK_SIZE)
 
-    # Additional token when using the overlap scheduler.
-    assert scheduler.build_connector_meta.call_count == NUM_TOKENS + int(
-        use_overlap_scheduler)
+    assert scheduler.build_connector_meta.call_count == NUM_TOKENS
 
     for i, call in enumerate(scheduler.build_connector_meta.call_args_list):
         sched_output = call.args[0]