code: def wait_on_job(self, job_id):
          
  
          while True:
              status = self.get_job_status(job_id)
              if status == 'SUCCEEDED':
                  logger.info('Batch job {} SUCCEEDED'.format(job_id))
                  return True
              elif status == 'FAILED':
                  # Raise and notify if job failed
                  jobs = self._client.describe_jobs(jobs=[job_id])['jobs']
                  job_str = json.dumps(jobs, indent=4)
                  logger.debug('Job details:\n' + job_str)
  
                  log_stream_name = jobs[0]['attempts'][0]['container']['logStreamName']
                  logs = self.get_logs(log_stream_name)
                  raise BatchJobException('Job {} failed: {}'.format(
                      job_id, logs))
  
              time.sleep(self.poll_time)
              logger.debug('Batch job status for job {0}: {1}'.format(
                  job_id, status))
docs: Poll task status until STOPPED