schrodinger.seam.testing.stu_tests.work_rebalancing module

Test that work rebalancing correctly redistributes elements from large in-progress bundles to idle workers.

Creates a pipeline where most elements are fast but one is very slow. The autoscaler spins up multiple workers, and idle workers should rebalance the remaining elements from the slow worker’s bundle.

Usage:

$SCHRODINGER/run python3 -m schrodinger.seam.testing.stu_tests.work_rebalancing

schrodinger.seam.testing.stu_tests.work_rebalancing.main()