Pulled common function get_equal_arm_allocations higher up.