From 2f9c5802aacaa975103740ffd7335c72ece7925d Mon Sep 17 00:00:00 2001 From: Vaib Suri Date: Tue, 22 Sep 2020 09:42:27 -0700 Subject: [PATCH] Fixing failing integ test due to new DescribeExecution response params --- tests/integ/test_training_pipeline_estimators.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/integ/test_training_pipeline_estimators.py b/tests/integ/test_training_pipeline_estimators.py index 2453b69..0607c57 100644 --- a/tests/integ/test_training_pipeline_estimators.py +++ b/tests/integ/test_training_pipeline_estimators.py @@ -119,6 +119,8 @@ def test_pca_estimator(sfn_client, sagemaker_session, sagemaker_role_arn, sfn_ro 'status': status, 'startDate': execution_info['startDate'], 'stopDate': execution_info['stopDate'], + 'inputDetails': {'included': True}, + 'outputDetails': {'included': True}, 'input': {'Training': {'AlgorithmSpecification': {'TrainingImage': estimator_image_uri, 'TrainingInputMode': 'File'}, 'OutputDataConfig': {'S3OutputPath': s3_output_path},