diff --git a/sagemaker-train/src/sagemaker/train/model_trainer.py b/sagemaker-train/src/sagemaker/train/model_trainer.py index d07edeb025..74b373b93e 100644 --- a/sagemaker-train/src/sagemaker/train/model_trainer.py +++ b/sagemaker-train/src/sagemaker/train/model_trainer.py @@ -326,7 +326,7 @@ def _populate_intelligent_defaults_from_training_job_space(self): if not self.networking: if default_enable_network_isolation is not None or default_vpc_config is not None: self.networking = Networking( - default_enable_network_isolation=default_enable_network_isolation, + enable_network_isolation=default_enable_network_isolation, subnets=self.config_mgr.resolve_value_from_config( config_path=TRAINING_JOB_SUBNETS_PATH ), @@ -342,8 +342,8 @@ def _populate_intelligent_defaults_from_training_job_space(self): config_path=TRAINING_JOB_SUBNETS_PATH ) if self.networking.security_group_ids is None: - self.networking.subnets = self.config_mgr.resolve_value_from_config( - config_path=TRAINING_JOB_SUBNETS_PATH + self.networking.security_group_ids = self.config_mgr.resolve_value_from_config( + config_path=TRAINING_JOB_SECURITY_GROUP_IDS_PATH ) if not self.output_data_config: