Skip to content

Commit 4dba051

Browse files
committed
test: widen ramp smoke timing margin
Signed-off-by: Eric W. Tramel <eric.tramel@gmail.com>
1 parent e8c9e49 commit 4dba051

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

packages/data-designer-engine/tests/engine/models/clients/test_throttled_model_client.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -508,12 +508,13 @@ async def _run_cold_server_scenario(
508508

509509
@pytest.mark.asyncio(loop_scope="session")
510510
async def test_startup_ramp_integration_eases_into_cold_server_without_429s() -> None:
511+
throttle_ramp_seconds = 0.3
511512
no_ramp_client, _, _ = await _run_cold_server_scenario(
512513
throttle_ramp_seconds=0.0,
513514
server_ramp_seconds=0.3,
514515
)
515516
ramped_client, throttle_manager, throttled_client = await _run_cold_server_scenario(
516-
throttle_ramp_seconds=0.3,
517+
throttle_ramp_seconds=throttle_ramp_seconds,
517518
server_ramp_seconds=0.1,
518519
)
519520

@@ -523,7 +524,7 @@ async def test_startup_ramp_integration_eases_into_cold_server_without_429s() ->
523524
assert ramped_client.peak_allowed > 1
524525
assert ramped_client.peak_in_flight > 1
525526

526-
await asyncio.sleep(0.31)
527+
await asyncio.sleep(throttle_ramp_seconds + 0.2)
527528
await _complete_with_rate_limit_retries(
528529
throttled_client,
529530
ChatCompletionRequest(model=MODEL_ID, messages=[{"role": "user", "content": "final ramp probe"}]),

0 commit comments

Comments
 (0)