def train_job_describe(sm_client, training_job_name):
  """API call to describe a training job."""
  try:
    response = sm_client.describe_training_job(
      TrainingJobName=training_job_name
    )
    return response
  except sm_client.exceptions.ResourceNotFound:
    raise Exception(f"Training job not found: '{training_job_name}'")