fix integ tests failing due to breaking ExecuteUserScriptError change

Author: Jonathan Esterhazy

tests/integ/test_chainer_train.py

@@ -115,7 +115,7 @@ def test_failed_training_job(sagemaker_session, chainer_full_version):
 
         with pytest.raises(ValueError) as e:
             chainer.fit()
-        assert 'This failure is expected' in str(e.value)
+        assert 'ExecuteUserScriptError' in str(e.value)
 
 
 def _run_mnist_training_job(sagemaker_session, instance_type, instance_count,

tests/integ/test_mxnet_train.py

@@ -112,4 +112,4 @@ def test_failed_training_job(sagemaker_session, mxnet_full_version):
 
         with pytest.raises(ValueError) as e:
             mx.fit()
-        assert 'This failure is expected' in str(e.value)
+        assert 'ExecuteUserScriptError' in str(e.value)

tests/integ/test_pytorch_train.py

@@ -109,7 +109,7 @@ def test_failed_training_job(sagemaker_session, pytorch_full_version):
 
         with pytest.raises(ValueError) as e:
             pytorch.fit()
-        assert 'This failure is expected' in str(e.value)
+        assert 'ExecuteUserScriptError' in str(e.value)
 
 
 def _upload_training_data(pytorch):

tests/integ/test_tf.py

@@ -162,4 +162,4 @@ def test_failed_tf_training(sagemaker_session, tf_full_version):
 
         with pytest.raises(ValueError) as e:
             estimator.fit()
-        assert 'This failure is expected' in str(e.value)
+        assert 'ExecuteUserScriptError' in str(e.value)