Esempi in Python per SessionCreateModelImageURIRenamer

Esempio n. 1

File: test_session_image_uri.py Progetto: 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)

Esempio n. 2

File: test_session_image_uri.py Progetto: 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()"))

Esempio n. 3

File: test_session_image_uri.py Progetto: 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)