What is the problem?

cdk SageMakerCreateTrainingJob class has no EnableNetworkIsolation option.

Without the attribute, no user can use AWS Marketplace model through CDK.

As doc says,

You can enable network isolation when you create your training job or model by setting the value of the EnableNetworkIsolation parameter to True when you call CreateTrainingJob, CreateHyperParameterTuningJob, or CreateModel.

and

Network isolation is required to run training jobs and models using resources from AWS Marketplace.

I tried to call an AWS Marketplace model without filling the attribute and got a following error.

EnableNetworkIsolation must be true for jobs using SageMaker Marketplace algorithms.
(Service: AmazonSageMaker; Status Code: 400; Error Code: ValidationException;
Request ID: xxxxxx; Proxy: null)

It looks CreateModel on master branch has the attribute, but not for CreateTrainingJob

Reproduction Steps

In python

SageMakerCreateTrainingJob(enable_network_isolation=True)

returns an error since the Class does not have such an attribute.

What did you expect to happen?

SageMakerCreateTrainingJob should accept enable_network_isolation attribute and can add the attribute to Cloud formation template.

What actually happened?

Since there's no such attribute, if you tried to use AWS Market place model, a user will be unable to call SageMakerCreateTrainingJob.

CDK CLI Version

1.125.0

Framework Version

No response

Node.js Version

14.18.0

OS

Lubuntu 20.10

Language

Typescript, Python

Language Version

No response

Other information

(It would be great if you tell me some workarounds.)