Otherwise, if for some reason NVIDIA operators aren't installed, `wait_for_slurm_cluster` hangs forever. But if we want to make `wait_for_slurm_cluster` depend on NVIDIA operators, we have to move it to a separate module.