Python _prepare_resource_config Examples

Example #1

File: test_job.py Project: zxy844288792/sagemaker-python-sdk

def test_prepare_resource_config():
    resource_config = _Job._prepare_resource_config(INSTANCE_COUNT, INSTANCE_TYPE, VOLUME_SIZE, None)

    assert resource_config == {
        'InstanceCount': INSTANCE_COUNT,
        'InstanceType': INSTANCE_TYPE,
        'VolumeSizeInGB': VOLUME_SIZE
    }

Example #2

File: test_job.py Project: zxy844288792/sagemaker-python-sdk

def test_prepare_resource_config_with_volume_kms():
    resource_config = _Job._prepare_resource_config(INSTANCE_COUNT, INSTANCE_TYPE, VOLUME_SIZE, VOLUME_KMS_KEY)

    assert resource_config == {
        'InstanceCount': INSTANCE_COUNT,
        'InstanceType': INSTANCE_TYPE,
        'VolumeSizeInGB': VOLUME_SIZE,
        'VolumeKmsKeyId': VOLUME_KMS_KEY
    }

Example #3

File: test_job.py Project: verdimrc/sagemaker-python-sdk

def test_prepare_resource_config():
    resource_config = _Job._prepare_resource_config(INSTANCE_COUNT,
                                                    INSTANCE_TYPE, VOLUME_SIZE,
                                                    None)

    assert resource_config == {
        "InstanceCount": INSTANCE_COUNT,
        "InstanceType": INSTANCE_TYPE,
        "VolumeSizeInGB": VOLUME_SIZE,
    }

Example #4

File: test_job.py Project: verdimrc/sagemaker-python-sdk

def test_prepare_resource_config_with_volume_kms():
    resource_config = _Job._prepare_resource_config(INSTANCE_COUNT,
                                                    INSTANCE_TYPE, VOLUME_SIZE,
                                                    VOLUME_KMS_KEY)

    assert resource_config == {
        "InstanceCount": INSTANCE_COUNT,
        "InstanceType": INSTANCE_TYPE,
        "VolumeSizeInGB": VOLUME_SIZE,
        "VolumeKmsKeyId": VOLUME_KMS_KEY,
    }

Example #5

File: test_job.py Project: Shuhuasong/Sage-Maker

def test_prepare_resource_config():
    resource_config = _Job._prepare_resource_config(INSTANCE_COUNT, INSTANCE_TYPE, VOLUME_SIZE)

    assert resource_config['InstanceCount'] == INSTANCE_COUNT
    assert resource_config['InstanceType'] == INSTANCE_TYPE
    assert resource_config['VolumeSizeInGB'] == VOLUME_SIZE