Remove hardcoded 'training' in error message for checking job status (#474)

Author: laurenyu

CHANGELOG.rst

@@ -13,6 +13,7 @@ CHANGELOG
 * build: added pylint
 * build: upgrade docker-compose to 1.23
 * enhancement: Frameworks: update warning for not setting framework_version as we aren't planning a breaking change anymore
+* enhancement: Session: remove hardcoded 'training' from job status error message
 
 1.14.1
 ======

src/sagemaker/session.py

@@ -636,7 +636,8 @@ def _check_job_status(self, job, desc, status_key_name):
 
         if status != 'Completed' and status != 'Stopped':
             reason = desc.get('FailureReason', '(No reason provided)')
-            raise ValueError('Error training {}: {} Reason: {}'.format(job, status, reason))
+            job_type = status_key_name.replace('JobStatus', ' job')
+            raise ValueError('Error for {} {}: {} Reason: {}'.format(job_type, job, status, reason))
 
     def wait_for_endpoint(self, endpoint, poll=5):
         """Wait for an Amazon SageMaker endpoint deployment to complete.