def METHOD_NAME(self):
    print("Waiting for training job to complete...")
    self.sagemaker_client.get_waiter("training_job_completed_or_stopped").wait(
        TrainingJobName=self.training_job_name
    )