code: def wait_for_job(self, job, poll=5):
          
          desc = _wait_until_training_done(lambda last_desc: _train_done(self.sagemaker_client, job, last_desc),
                                           None, poll)
          self._check_job_status(job, desc, 'TrainingJobStatus')
          return desc
docs: Wait for an Amazon SageMaker training job to complete.
  
          Args:
              job (str): Name of the training job to wait for.
              poll (int): Polling interval in seconds (default: 5).
  
          Returns:
              (dict): Return value from the ``DescribeTrainingJob`` API.
  
          Raises:
              ValueError: If the training job fails.