Exemplos de Model.delete_model em Python

Exemplo n.º 1

Arquivo: test_model.py Projeto: jkroll-aws/sagemaker-python-sdk

def test_delete_model(sagemaker_session):
    model = Model(MODEL_IMAGE,
                  MODEL_DATA,
                  name=MODEL_NAME,
                  sagemaker_session=sagemaker_session)

    model.delete_model()
    sagemaker_session.delete_model.assert_called_with(model.name)

Exemplo n.º 2

def test_delete_model_no_name(sagemaker_session):
    model = Model(MODEL_IMAGE, MODEL_DATA, sagemaker_session=sagemaker_session)

    with pytest.raises(
        ValueError, match="The SageMaker model must be created first before attempting to delete."
    ):
        model.delete_model()
    sagemaker_session.delete_model.assert_not_called()