Exemplos de SessionCreateModelImageURIRenamer em Python

Exemplo n.º 1

Arquivo: test_session_image_uri.py Projeto: xiaoyi-cheng/sagemaker-python-sdk

def test_create_model_modify_node():
    modifier = renamed_params.SessionCreateModelImageURIRenamer()

    for template in CREATE_MODEL_TEMPLATES:
        call = ast_call(template.format("primary_container_image=my_image"))
        modifier.modify_node(call)

        expected = template.format("image_uri=my_image")
        assert expected == pasta.dump(call)

Exemplo n.º 2

Arquivo: test_session_image_uri.py Projeto: xiaoyi-cheng/sagemaker-python-sdk

def test_create_model_node_should_be_modified_random_function_call():
    modifier = renamed_params.SessionCreateModelImageURIRenamer()
    assert not modifier.node_should_be_modified(ast_call("create_model()"))

Exemplo n.º 3

Arquivo: test_session_image_uri.py Projeto: xiaoyi-cheng/sagemaker-python-sdk

def test_create_model_node_should_be_modified_no_image():
    modifier = renamed_params.SessionCreateModelImageURIRenamer()

    for template in CREATE_MODEL_TEMPLATES:
        call = ast_call(template.format(""))
        assert not modifier.node_should_be_modified(call)