Python _format_inputs_to_input_config примеры использования

Пример #1

Файл: test_transformer.py Проект: ngluna/sagemaker-python-sdk

def test_format_inputs_to_input_config():
    expected_config = {"DataSource": {"S3DataSource": {"S3DataType": S3_DATA_TYPE, "S3Uri": DATA}}}

    actual_config = _TransformJob._format_inputs_to_input_config(
        DATA, S3_DATA_TYPE, None, None, None
    )
    assert actual_config == expected_config

Пример #2

Файл: test_transformer.py Проект: xuefeng7/sagemaker-python-sdk

def test_format_inputs_to_input_config():
    expected_config = {
        'DataSource': {
            'S3DataSource': {
                'S3DataType': S3_DATA_TYPE,
                'S3Uri': DATA,
            },
        },
    }

    actual_config = _TransformJob._format_inputs_to_input_config(DATA, S3_DATA_TYPE, None, None, None)
    assert actual_config == expected_config

Пример #3

Файл: test_transformer.py Проект: cheesama/sagemaker-python-sdk

def test_format_inputs_to_input_config():
    expected_config = {
        'DataSource': {
            'S3DataSource': {
                'S3DataType': S3_DATA_TYPE,
                'S3Uri': DATA,
            },
        },
    }

    actual_config = _TransformJob._format_inputs_to_input_config(DATA, S3_DATA_TYPE, None, None, None)
    assert actual_config == expected_config

Пример #4

Файл: test_transformer.py Проект: NiweditaJ19/sagemaker-python-sdk

def test_format_inputs_to_input_config_with_optional_params():
    compression = "Gzip"
    content_type = "text/csv"
    split = "Line"

    expected_config = {
        "DataSource": {"S3DataSource": {"S3DataType": S3_DATA_TYPE, "S3Uri": DATA}},
        "CompressionType": compression,
        "ContentType": content_type,
        "SplitType": split,
    }

    actual_config = _TransformJob._format_inputs_to_input_config(
        DATA, S3_DATA_TYPE, content_type, compression, split
    )
    assert actual_config == expected_config

Пример #5

Файл: test_transformer.py Проект: xuefeng7/sagemaker-python-sdk

def test_format_inputs_to_input_config_with_optional_params():
    compression = 'Gzip'
    content_type = 'text/csv'
    split = 'Line'

    expected_config = {
        'DataSource': {
            'S3DataSource': {
                'S3DataType': S3_DATA_TYPE,
                'S3Uri': DATA,
            },
        },
        'CompressionType': compression,
        'ContentType': content_type,
        'SplitType': split,
    }

    actual_config = _TransformJob._format_inputs_to_input_config(DATA, S3_DATA_TYPE, content_type, compression, split)
    assert actual_config == expected_config

Пример #6

Файл: test_transformer.py Проект: cheesama/sagemaker-python-sdk

def test_format_inputs_to_input_config_with_optional_params():
    compression = 'Gzip'
    content_type = 'text/csv'
    split = 'Line'

    expected_config = {
        'DataSource': {
            'S3DataSource': {
                'S3DataType': S3_DATA_TYPE,
                'S3Uri': DATA,
            },
        },
        'CompressionType': compression,
        'ContentType': content_type,
        'SplitType': split,
    }

    actual_config = _TransformJob._format_inputs_to_input_config(DATA, S3_DATA_TYPE, content_type, compression, split)
    assert actual_config == expected_config