Python run_api_server_docker_containerの例

コード例 #1

ファイル: conftest.py プロジェクト: zidingz/BentoML

def host(clean_context, test_svc_bundle, enable_microbatch, with_docker):
    if with_docker:
        image = clean_context.enter_context(
            build_api_server_docker_image(test_svc_bundle, "example_service"))
        with run_api_server_docker_container(image, enable_microbatch) as host:
            yield host
    else:
        with run_api_server(test_svc_bundle, enable_microbatch) as host:
            yield host

コード例 #2

ファイル: conftest.py プロジェクト: gorbunov-ka-nlmk/BentoML_testA

def host(pytestconfig, clean_context, enable_microbatch):
    test_svc_bundle = pytestconfig.getoption("bento_dist") or os.path.join(
        sys.argv[1], "build", "dist")

    if pytestconfig.getoption("docker"):
        image = clean_context.enter_context(
            build_api_server_docker_image(test_svc_bundle, "example_service"))
        with run_api_server_docker_container(image, enable_microbatch) as host:
            yield host
    else:
        with run_api_server(test_svc_bundle, enable_microbatch) as host:
            yield host

コード例 #3

def host(pytestconfig, clean_context):
    """
    Launch host from a
    """
    test_svc_bundle = pytestconfig.getoption("bento_dist") or os.path.join(
        sys.argv[2], "build", "dist")

    config_file = pytestconfig.getoption("config_file") or os.path.join(
        sys.argv[2], "bentoml_config.yml")
    if not os.path.exists(config_file):
        if pytestconfig.getoption("config_file"):
            raise Exception(f"config file not found: {config_file}")
        else:
            config_file = None

    if pytestconfig.getoption("docker"):
        image = clean_context.enter_context(
            build_api_server_docker_image(test_svc_bundle, "example_service"))
        with run_api_server_docker_container(
                image,
                config_file=config_file,
        ) as host:
            yield host
    elif pytestconfig.getoption("dev_server"):
        with run_api_server(
                test_svc_bundle,
                config_file=config_file,
                dev_server=True,
        ) as host:
            yield host
    else:
        with run_api_server(
                test_svc_bundle,
                config_file=config_file,
        ) as host:
            yield host

コード例 #4

ファイル: test_h2o_model_artifact.py プロジェクト: Nikunj3masarani/bentoml

def h2o_docker_host(h2o_image):
    with run_api_server_docker_container(h2o_image, timeout=500) as host:
        yield host

コード例 #5

ファイル: test_tensorflow_v2_2_savedmodel_artifact.py プロジェクト: Nikunj3masarani/bentoml

def host(image, enable_microbatch):
    with run_api_server_docker_container(
        image, enable_microbatch=enable_microbatch, timeout=500
    ) as host:
        yield host

コード例 #6

ファイル: test_xgboost_model_artifact.py プロジェクト: subhayuroy/BentoML

def xgboost_docker_host(xgboost_image):
    with run_api_server_docker_container(xgboost_image, timeout=60) as host:
        yield host

コード例 #7

def xgboost_docker_host(xgboost_image):
    with run_api_server_docker_container(xgboost_image,
                                         enable_microbatch=False,
                                         timeout=60) as host:
        yield host

コード例 #8

def transformers_host(transformers_image):
    with run_api_server_docker_container(transformers_image,
                                         timeout=500) as host:
        yield host