How to use the sagemaker.fw_registry.registry function in sagemaker

To help you get started, we’ve selected a few sagemaker examples, based on popular ways it is used in public projects.

Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.

github aws / sagemaker-python-sdk / tests / unit / test_fw_registry.py View on Github external
def test_registry_sklearn():
    assert (
        registry("us-west-1", scikit_learn_framework_name)
        == "746614075791.dkr.ecr.us-west-1.amazonaws.com"
    )
    assert (
        registry("us-west-2", scikit_learn_framework_name)
        == "246618743249.dkr.ecr.us-west-2.amazonaws.com"
    )
    assert (
        registry("us-east-1", scikit_learn_framework_name)
        == "683313688378.dkr.ecr.us-east-1.amazonaws.com"
    )
    assert (
        registry("us-east-2", scikit_learn_framework_name)
        == "257758044811.dkr.ecr.us-east-2.amazonaws.com"
    )
    assert (
        registry("ap-northeast-1", scikit_learn_framework_name)
        == "354813040037.dkr.ecr.ap-northeast-1.amazonaws.com"
    )
    assert (
        registry("ap-northeast-2", scikit_learn_framework_name)
        == "366743142698.dkr.ecr.ap-northeast-2.amazonaws.com"
    )
    assert (
        registry("ap-southeast-1", scikit_learn_framework_name)
        == "121021644041.dkr.ecr.ap-southeast-1.amazonaws.com"
    )
    assert (
        registry("ap-southeast-2", scikit_learn_framework_name)
github aws / sagemaker-python-sdk / tests / unit / test_fw_registry.py View on Github external
)
    assert (
        registry("ap-southeast-2", "sparkml-serving")
        == "783357654285.dkr.ecr.ap-southeast-2.amazonaws.com"
    )
    assert (
        registry("ap-south-1", "sparkml-serving") == "720646828776.dkr.ecr.ap-south-1.amazonaws.com"
    )
    assert (
        registry("eu-west-1", "sparkml-serving") == "141502667606.dkr.ecr.eu-west-1.amazonaws.com"
    )
    assert (
        registry("eu-west-2", "sparkml-serving") == "764974769150.dkr.ecr.eu-west-2.amazonaws.com"
    )
    assert (
        registry("eu-central-1", "sparkml-serving")
        == "492215442770.dkr.ecr.eu-central-1.amazonaws.com"
    )
    assert (
        registry("ca-central-1", "sparkml-serving")
        == "341280168497.dkr.ecr.ca-central-1.amazonaws.com"
    )
    assert (
        registry("us-gov-west-1", "sparkml-serving")
        == "414596584902.dkr.ecr.us-gov-west-1.amazonaws.com"
    )
    assert (
        registry("us-iso-east-1", "sparkml-serving")
        == "833128469047.dkr.ecr.us-iso-east-1.c2s.ic.gov"
    )
github aws / sagemaker-python-sdk / tests / unit / test_fw_registry.py View on Github external
def test_registry_sparkml_serving():
    assert (
        registry("us-west-1", "sparkml-serving") == "746614075791.dkr.ecr.us-west-1.amazonaws.com"
    )
    assert (
        registry("us-west-2", "sparkml-serving") == "246618743249.dkr.ecr.us-west-2.amazonaws.com"
    )
    assert (
        registry("us-east-1", "sparkml-serving") == "683313688378.dkr.ecr.us-east-1.amazonaws.com"
    )
    assert (
        registry("us-east-2", "sparkml-serving") == "257758044811.dkr.ecr.us-east-2.amazonaws.com"
    )
    assert (
        registry("ap-northeast-1", "sparkml-serving")
        == "354813040037.dkr.ecr.ap-northeast-1.amazonaws.com"
    )
    assert (
        registry("ap-northeast-2", "sparkml-serving")
        == "366743142698.dkr.ecr.ap-northeast-2.amazonaws.com"
    )
    assert (
github aws / sagemaker-python-sdk / tests / unit / test_fw_registry.py View on Github external
def test_region_invalid():
    with pytest.raises(KeyError):
        registry("us-west-5", "scikit-learn")
github aws / sagemaker-python-sdk / tests / unit / test_fw_registry.py View on Github external
registry("ap-northeast-2", "sparkml-serving")
        == "366743142698.dkr.ecr.ap-northeast-2.amazonaws.com"
    )
    assert (
        registry("ap-southeast-1", "sparkml-serving")
        == "121021644041.dkr.ecr.ap-southeast-1.amazonaws.com"
    )
    assert (
        registry("ap-southeast-2", "sparkml-serving")
        == "783357654285.dkr.ecr.ap-southeast-2.amazonaws.com"
    )
    assert (
        registry("ap-south-1", "sparkml-serving") == "720646828776.dkr.ecr.ap-south-1.amazonaws.com"
    )
    assert (
        registry("eu-west-1", "sparkml-serving") == "141502667606.dkr.ecr.eu-west-1.amazonaws.com"
    )
    assert (
        registry("eu-west-2", "sparkml-serving") == "764974769150.dkr.ecr.eu-west-2.amazonaws.com"
    )
    assert (
        registry("eu-central-1", "sparkml-serving")
        == "492215442770.dkr.ecr.eu-central-1.amazonaws.com"
    )
    assert (
        registry("ca-central-1", "sparkml-serving")
        == "341280168497.dkr.ecr.ca-central-1.amazonaws.com"
    )
    assert (
        registry("us-gov-west-1", "sparkml-serving")
        == "414596584902.dkr.ecr.us-gov-west-1.amazonaws.com"
    )
github aws / sagemaker-python-sdk / tests / unit / test_fw_registry.py View on Github external
def test_registry_sklearn():
    assert (
        registry("us-west-1", scikit_learn_framework_name)
        == "746614075791.dkr.ecr.us-west-1.amazonaws.com"
    )
    assert (
        registry("us-west-2", scikit_learn_framework_name)
        == "246618743249.dkr.ecr.us-west-2.amazonaws.com"
    )
    assert (
        registry("us-east-1", scikit_learn_framework_name)
        == "683313688378.dkr.ecr.us-east-1.amazonaws.com"
    )
    assert (
        registry("us-east-2", scikit_learn_framework_name)
        == "257758044811.dkr.ecr.us-east-2.amazonaws.com"
    )
    assert (
        registry("ap-northeast-1", scikit_learn_framework_name)
github aws / sagemaker-python-sdk / tests / unit / test_fw_registry.py View on Github external
== "683313688378.dkr.ecr.us-east-1.amazonaws.com"
    )
    assert (
        registry("us-east-2", scikit_learn_framework_name)
        == "257758044811.dkr.ecr.us-east-2.amazonaws.com"
    )
    assert (
        registry("ap-northeast-1", scikit_learn_framework_name)
        == "354813040037.dkr.ecr.ap-northeast-1.amazonaws.com"
    )
    assert (
        registry("ap-northeast-2", scikit_learn_framework_name)
        == "366743142698.dkr.ecr.ap-northeast-2.amazonaws.com"
    )
    assert (
        registry("ap-southeast-1", scikit_learn_framework_name)
        == "121021644041.dkr.ecr.ap-southeast-1.amazonaws.com"
    )
    assert (
        registry("ap-southeast-2", scikit_learn_framework_name)
        == "783357654285.dkr.ecr.ap-southeast-2.amazonaws.com"
    )
    assert (
        registry("ap-south-1", scikit_learn_framework_name)
        == "720646828776.dkr.ecr.ap-south-1.amazonaws.com"
    )
    assert (
        registry("eu-west-1", scikit_learn_framework_name)
        == "141502667606.dkr.ecr.eu-west-1.amazonaws.com"
    )
    assert (
        registry("eu-west-2", scikit_learn_framework_name)
github aws / sagemaker-python-sdk / tests / unit / test_fw_registry.py View on Github external
def test_registry_sparkml_serving():
    assert (
        registry("us-west-1", "sparkml-serving") == "746614075791.dkr.ecr.us-west-1.amazonaws.com"
    )
    assert (
        registry("us-west-2", "sparkml-serving") == "246618743249.dkr.ecr.us-west-2.amazonaws.com"
    )
    assert (
        registry("us-east-1", "sparkml-serving") == "683313688378.dkr.ecr.us-east-1.amazonaws.com"
    )
    assert (
        registry("us-east-2", "sparkml-serving") == "257758044811.dkr.ecr.us-east-2.amazonaws.com"
    )
    assert (
        registry("ap-northeast-1", "sparkml-serving")
        == "354813040037.dkr.ecr.ap-northeast-1.amazonaws.com"
    )
    assert (
        registry("ap-northeast-2", "sparkml-serving")
github aws / sagemaker-python-sdk / tests / unit / test_sparkml_serving.py View on Github external
def test_sparkml_model(sagemaker_session):
    sparkml = SparkMLModel(sagemaker_session=sagemaker_session, model_data=MODEL_DATA, role=ROLE)
    assert sparkml.image == registry(REGION, "sparkml-serving") + "/sagemaker-sparkml-serving:2.2"
github aws / sagemaker-python-sdk / tests / unit / test_fw_registry.py View on Github external
== "366743142698.dkr.ecr.ap-northeast-2.amazonaws.com"
    )
    assert (
        registry("ap-southeast-1", scikit_learn_framework_name)
        == "121021644041.dkr.ecr.ap-southeast-1.amazonaws.com"
    )
    assert (
        registry("ap-southeast-2", scikit_learn_framework_name)
        == "783357654285.dkr.ecr.ap-southeast-2.amazonaws.com"
    )
    assert (
        registry("ap-south-1", scikit_learn_framework_name)
        == "720646828776.dkr.ecr.ap-south-1.amazonaws.com"
    )
    assert (
        registry("eu-west-1", scikit_learn_framework_name)
        == "141502667606.dkr.ecr.eu-west-1.amazonaws.com"
    )
    assert (
        registry("eu-west-2", scikit_learn_framework_name)
        == "764974769150.dkr.ecr.eu-west-2.amazonaws.com"
    )
    assert (
        registry("eu-central-1", scikit_learn_framework_name)
        == "492215442770.dkr.ecr.eu-central-1.amazonaws.com"
    )
    assert (
        registry("ca-central-1", scikit_learn_framework_name)
        == "341280168497.dkr.ecr.ca-central-1.amazonaws.com"
    )
    assert (
        registry("us-gov-west-1", scikit_learn_framework_name)