schrodinger.seam.testing.stu_tests.streamed_bundles_trial_rate module¶

Test that streamed bundles trial fleet correctly estimates worker count.

Verifies two behaviors with SCHRODINGER_USE_STREAMED_BUNDLES enabled:

When elements complete during the trial, the executor autoscales to fewer workers than the maximum based on the measured rate.
When elements are too slow to complete during the trial (including the finish_bundle grace period), the executor falls back to the maximum number of workers.

Usage:

SCHRODINGER_USE_STREAMED_BUNDLES=True \
    $SCHRODINGER/run python3 -m schrodinger.seam.testing.stu_tests.streamed_bundles_trial_rate

schrodinger.seam.testing.stu_tests.streamed_bundles_trial_rate.test_autoscale_below_max()¶: 100 elements x 0.5s each. Trial measures ~2 elem/s, total work ~50s. Autoscaling should pick far fewer than MAX_WORKERS.

schrodinger.seam.testing.stu_tests.streamed_bundles_trial_rate.test_fallback_to_max()¶: 10 elements x 15s each. Trial walltime 5s, grace periods 5s each. Element can’t complete within trial + grace, so no rate data. Executor should fall back to max workers.