Skip to content

Commit 4d8b871

Browse files
committed
fix use input_data_config parameter in train as authoritative source
1 parent 6c750be commit 4d8b871

File tree

1 file changed

+2
-11
lines changed

1 file changed

+2
-11
lines changed

src/sagemaker/modules/train/model_trainer.py

Lines changed: 2 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -596,17 +596,8 @@ def train(
596596
self._populate_intelligent_defaults()
597597
current_training_job_name = _get_unique_name(self.base_job_name)
598598
input_data_key_prefix = f"{self.base_job_name}/{current_training_job_name}/input"
599-
if input_data_config and self.input_data_config:
600-
final_channels = {
601-
input_data.channel_name: input_data for input_data in self.input_data_config
602-
}
603-
# Update with precedence on the input_data_config passed into the train method
604-
final_channels.update(
605-
{input_data.channel_name: input_data for input_data in input_data_config}
606-
)
607-
self.input_data_config = list(final_channels.values())
608-
else:
609-
self.input_data_config = input_data_config or self.input_data_config or []
599+
600+
self.input_data_config = input_data_config or self.input_data_config or []
610601

611602
if self.input_data_config:
612603
input_data_config = self._get_input_data_config(

0 commit comments

Comments
 (0)