File tree Expand file tree Collapse file tree 1 file changed +3
-12
lines changed
src/sagemaker/modules/train Expand file tree Collapse file tree 1 file changed +3
-12
lines changed Original file line number Diff line number Diff line change @@ -596,20 +596,11 @@ 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 :
612- input_data_config = self ._get_input_data_config (
603+ self . input_data_config = self ._get_input_data_config (
613604 self .input_data_config , input_data_key_prefix
614605 )
615606
You can’t perform that action at this time.
0 commit comments