def initialize()

in src/sagemaker_xgboost_container/algorithm_mode/channel_validation.py [0:0]

• 20 lines of code
• 6 McCabe index (conditional complexity)

def initialize():
    train_channel = cv.Channel(name="train", required=True)
    for ct in VALID_CONTENT_TYPES:
        train_channel.add(ct, cv.Channel.FILE_MODE, cv.Channel.SHARDED)
        train_channel.add(ct, cv.Channel.FILE_MODE, cv.Channel.REPLICATED)

    for ct in VALID_PIPED_CONTENT_TYPES:
        train_channel.add(ct, cv.Channel.PIPE_MODE, cv.Channel.SHARDED)
        train_channel.add(ct, cv.Channel.PIPE_MODE, cv.Channel.REPLICATED)

    validation_channel = cv.Channel(name="validation", required=False)
    for ct in VALID_CONTENT_TYPES:
        validation_channel.add(ct, cv.Channel.FILE_MODE, cv.Channel.SHARDED)
        validation_channel.add(ct, cv.Channel.FILE_MODE, cv.Channel.REPLICATED)

    for ct in VALID_PIPED_CONTENT_TYPES:
        validation_channel.add(ct, cv.Channel.PIPE_MODE, cv.Channel.SHARDED)
        validation_channel.add(ct, cv.Channel.PIPE_MODE, cv.Channel.REPLICATED)

    # new for script mode/algorithm mode toggle
    code_channel = cv.Channel(name="code", required=False)
    code_channel.add("text/python", cv.Channel.FILE_MODE, cv.Channel.REPLICATED)

    data_channels = cv.Channels(train_channel, validation_channel, code_channel)
    data_channels.set_default_content_type("text/libsvm")

    return data_channels