Fix security_group_ids

admivsn · admivsn · commit f08349d064f6 · 2026-04-16T16:36:00.000Z
diff --git a/sagemaker-train/src/sagemaker/train/model_trainer.py b/sagemaker-train/src/sagemaker/train/model_trainer.py
@@ -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:

Original file line number	Diff line number	Diff line change
`@@ -342,8 +342,8 @@ def _populate_intelligent_defaults_from_training_job_space(self):`
`342`	`342`	`config_path=TRAINING_JOB_SUBNETS_PATH`
`343`	`343`	`)`
`344`	`344`	`if self.networking.security_group_ids is None:`
`345`		`- self.networking.subnets = self.config_mgr.resolve_value_from_config(`
`346`		`- config_path=TRAINING_JOB_SUBNETS_PATH`
	`345`	`+ self.networking.security_group_ids = self.config_mgr.resolve_value_from_config(`
	`346`	`+ config_path=TRAINING_JOB_SECURITY_GROUP_IDS_PATH`
`347`	`347`	`)`
`348`	`348`
`349`	`349`	`if not self.output_data_config:`