Pass network isolation config to pipelineModel

shreyapandit · shreyapandit · commit 2fa31c283f2d · 2021-08-19T08:40:37.000-07:00
diff --git a/src/sagemaker/automl/automl.py b/src/sagemaker/automl/automl.py
@@ -328,6 +328,7 @@ def create_model(
             predictor_cls=predictor_cls,
             name=name,
             vpc_config=vpc_config,
+            enable_network_isolation=enable_network_isolation,
             sagemaker_session=sagemaker_session or self.sagemaker_session,
         )
         return pipeline

Original file line number	Diff line number	Diff line change
`@@ -328,6 +328,7 @@ def create_model(`
`328`	`328`	`predictor_cls=predictor_cls,`
`329`	`329`	`name=name,`
`330`	`330`	`vpc_config=vpc_config,`
	`331`	`+ enable_network_isolation=enable_network_isolation,`
`331`	`332`	`sagemaker_session=sagemaker_session or self.sagemaker_session,`
`332`	`333`	`)`
`333`	`334`	`return pipeline`