fix(aws-stepfunctions-tasks): SageMaker create training job has incor…

…rect property name for AttributeNames (#10026) Rename mistyped property `AtttributeNames` to `AttributeNames` Fixes #10014 ---- *By submitting this pull request, I confirm that my contribution is made under the terms of the Apache-2.0 license*
aws · Aug 28, 2020 · ba51ea3 · ba51ea3
1 parent 6114045
commit ba51ea3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/packages/@aws-cdk/aws-stepfunctions-tasks/lib/sagemaker/create-training-job.ts b/packages/@aws-cdk/aws-stepfunctions-tasks/lib/sagemaker/create-training-job.ts
@@ -253,7 +253,7 @@ export class SageMakerCreateTrainingJob extends sfn.TaskStateBase implements iam
             ...(channel.dataSource.s3DataSource.s3DataDistributionType
               ? { S3DataDistributionType: channel.dataSource.s3DataSource.s3DataDistributionType }
               : {}),
-            ...(channel.dataSource.s3DataSource.attributeNames ? { AtttributeNames: channel.dataSource.s3DataSource.attributeNames } : {}),
+            ...(channel.dataSource.s3DataSource.attributeNames ? { AttributeNames: channel.dataSource.s3DataSource.attributeNames } : {}),
           },
         },
         ...(channel.compressionType ? { CompressionType: channel.compressionType } : {}),

diff --git a/packages/@aws-cdk/aws-stepfunctions-tasks/test/sagemaker/create-training-job.test.ts b/packages/@aws-cdk/aws-stepfunctions-tasks/test/sagemaker/create-training-job.test.ts
@@ -153,6 +153,7 @@ test('create complex training job', () => {
         recordWrapperType: tasks.RecordWrapperType.RECORD_IO,
         dataSource: {
           s3DataSource: {
+            attributeNames: ['source-ref', 'class'],
             s3DataType: tasks.S3DataType.S3_PREFIX,
             s3Location: tasks.S3Location.fromBucket(s3.Bucket.fromBucketName(stack, 'InputBucketA', 'mybucket'), 'mytrainpath'),
           },
@@ -165,6 +166,7 @@ test('create complex training job', () => {
         recordWrapperType: tasks.RecordWrapperType.RECORD_IO,
         dataSource: {
           s3DataSource: {
+            attributeNames: ['source-ref', 'class'],
             s3DataType: tasks.S3DataType.S3_PREFIX,
             s3Location: tasks.S3Location.fromBucket(s3.Bucket.fromBucketName(stack, 'InputBucketB', 'mybucket'), 'mytestpath'),
           },
@@ -230,6 +232,7 @@ test('create complex training job', () => {
           ContentType: 'image/jpeg',
           DataSource: {
             S3DataSource: {
+              AttributeNames: ['source-ref', 'class'],
               S3DataType: 'S3Prefix',
               S3Uri: {
                 'Fn::Join': ['', ['https://s3.', { Ref: 'AWS::Region' }, '.', { Ref: 'AWS::URLSuffix' }, '/mybucket/mytrainpath']],
@@ -244,6 +247,7 @@ test('create complex training job', () => {
           ContentType: 'image/jpeg',
           DataSource: {
             S3DataSource: {
+              AttributeNames: ['source-ref', 'class'],
               S3DataType: 'S3Prefix',
               S3Uri: {
                 'Fn::Join': ['', ['https://s3.', { Ref: 'AWS::Region' }, '.', { Ref: 'AWS::URLSuffix' }, '/mybucket/mytestpath']],