Ejemplos de EstimatorCreateModelImageURIRenamer en Python

Ejemplo n.º 1

Archivo: test_estimator_create_model.py Proyecto: zhangchi1/sagemaker-python-sdk

def test_modify_node():
    node = ast_call("estimator.create_model(image=my_image)")
    modifier = renamed_params.EstimatorCreateModelImageURIRenamer()
    modifier.modify_node(node)

    expected = "estimator.create_model(image_uri=my_image)"
    assert expected == pasta.dump(node)

Ejemplo n.º 2

Archivo: test_estimator_create_model.py Proyecto: zhangchi1/sagemaker-python-sdk

def test_node_should_be_modified_no_distribution():
    modifier = renamed_params.EstimatorCreateModelImageURIRenamer()

    for estimator in ESTIMATORS:
        call = "{}.create_model()".format(estimator)
        assert not modifier.node_should_be_modified(ast_call(call))

Ejemplo n.º 3

Archivo: test_estimator_create_model.py Proyecto: zhangchi1/sagemaker-python-sdk

def test_node_should_be_modified_random_function_call():
    modifier = renamed_params.EstimatorCreateModelImageURIRenamer()
    assert not modifier.node_should_be_modified(ast_call("create_model()"))

Ejemplo n.º 4

Archivo: test_estimator_create_model.py Proyecto: zhangchi1/sagemaker-python-sdk

def test_node_should_be_modified():
    modifier = renamed_params.EstimatorCreateModelImageURIRenamer()

    for estimator in ESTIMATORS:
        call = "{}.create_model(image='my-image:latest')".format(estimator)
        assert modifier.node_should_be_modified(ast_call(call))