fix condition

benieric · benieric · commit d301784f194a · 2025-05-14T11:28:43.000-07:00
diff --git a/src/sagemaker/modules/train/model_trainer.py b/src/sagemaker/modules/train/model_trainer.py
@@ -529,7 +529,7 @@ def model_post_init(self, __context: Any):
                             "Must be a valid JSON or YAML file."
                         )
 
-        if self.training_mode == Mode.SAGEMAKER_TRAINING_JOB and self.output_data_config is None:
+        if self.training_mode == Mode.SAGEMAKER_TRAINING_JOB:
             if self.output_data_config is None:
                 session = self.sagemaker_session
                 base_job_name = self.base_job_name

Original file line number	Diff line number	Diff line change
`@@ -529,7 +529,7 @@ def model_post_init(self, __context: Any):`
`529`	`529`	`"Must be a valid JSON or YAML file."`
`530`	`530`	`)`
`531`	`531`
`532`		`- if self.training_mode == Mode.SAGEMAKER_TRAINING_JOB and self.output_data_config is None:`
	`532`	`+ if self.training_mode == Mode.SAGEMAKER_TRAINING_JOB:`
`533`	`533`	`if self.output_data_config is None:`
`534`	`534`	`session = self.sagemaker_session`
`535`	`535`	`base_job_name = self.base_job_name`