def _validate_input_data_config()

in src/sagemaker_tensorflow/pipemode.py [0:0]

• 5 lines of code
• 3 McCabe index (conditional complexity)

    def _validate_input_data_config(self):
        if self.channel not in self.input_data_config:
            raise PipeModeDatasetException("Channel {} not found in Training Job InputDataConfig".format(self.channel))
        if self.input_data_config[self.channel].get('TrainingInputMode', "").lower() != "pipe":
            raise PipeModeDatasetException("Channel {} is not a PipeMode channel".format(self.channel))