Python _prepare_resource_config Beispiele

Beispiel #1

Datei: test_job.py Projekt: 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
    }

Beispiel #2

Datei: test_job.py Projekt: 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
    }

Beispiel #3

Datei: test_job.py Projekt: 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,
    }

Beispiel #4

Datei: test_job.py Projekt: 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,
    }

Beispiel #5

Datei: test_job.py Projekt: 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