Esempio n. 1
0
def test_undeploy_pipeline_docker(inference_pipeline: Pipeline,
                                  docker_runtime: SeldonDockerRuntime):
    inference_pipeline.undeploy()

    for model in inference_pipeline._models:
        with pytest.raises(docker.errors.NotFound):
            docker_runtime._get_container(model._details)
Esempio n. 2
0
def test_undeploy_pipeline_docker(inference_pipeline: Pipeline,
                                  runtime: SeldonDockerRuntime):
    runtime.undeploy(inference_pipeline)

    for model in inference_pipeline.models.values():
        with pytest.raises(docker.errors.NotFound):
            runtime._get_container(model.model_spec)
Esempio n. 3
0
def test_undeploy_docker(sklearn_model: Model, runtime: SeldonDockerRuntime):
    time.sleep(2)

    runtime.undeploy(sklearn_model)

    with pytest.raises(docker.errors.NotFound):
        runtime._get_container(sklearn_model.model_spec)
Esempio n. 4
0
def test_undeploy_docker(sklearn_model: Model, docker_runtime: SeldonDockerRuntime):
    sklearn_model.deploy()
    time.sleep(2)

    sklearn_model.undeploy()

    with pytest.raises(docker.errors.NotFound):
        docker_runtime._get_container(sklearn_model.details)
Esempio n. 5
0
def test_deploy_pipeline_docker(
    inference_pipeline: Pipeline,
    runtime: SeldonDockerRuntime,
):
    for model in inference_pipeline.models.values():
        container = runtime._get_container(model.model_spec)
        assert container.status == "running"

    pipeline_container = runtime._get_container(inference_pipeline.model_spec)
    assert pipeline_container.status == "running"
Esempio n. 6
0
def test_deploy_pipeline_docker(
    inference_pipeline: Pipeline,
    docker_runtime: SeldonDockerRuntime,
    docker_runtime_v2: SeldonDockerRuntime,
):
    for model in inference_pipeline._models:
        container = docker_runtime._get_container(model.details)
        assert container.status == "running"

    pipeline_container = docker_runtime_v2._get_container(inference_pipeline.details)
    assert pipeline_container.status == "running"
Esempio n. 7
0
def test_deploy_docker(sklearn_model: Model, docker_runtime: SeldonDockerRuntime):
    sklearn_model.deploy()
    time.sleep(2)

    container = docker_runtime._get_container(sklearn_model.details)
    assert container.status == "running"

    sklearn_model.undeploy()
Esempio n. 8
0
def test_launch_tensorflow(cifar10_model: Model, runtime: SeldonDockerRuntime):
    container = runtime._get_container(cifar10_model.model_spec)
    assert container.status == "running"
Esempio n. 9
0
def test_deploy_docker(sklearn_model_deployed, runtime: SeldonDockerRuntime):

    time.sleep(2)

    container = runtime._get_container(sklearn_model_deployed.model_spec)
    assert container.status == "running"