Restore waiting for local jobs
Restore waiting for local jobs

PiperOrigin-RevId: 398711124
Change-Id: I1147bde9de9cc53b9684eaa1de1dddbd57e69709
GitOrigin-RevId: 6b45dc14e454b55fef81704d0d29671de963c656

deepmind/xmanager,deepmind/xmanager