def tf1_host(tf1_svc, enable_microbatch, clean_context):
with export_service_bundle(tf1_svc) as saved_path:
server_image = clean_context.enter_context(
build_api_server_docker_image(saved_path))
with run_api_server_docker_container(
server_image, enable_microbatch=enable_microbatch,
timeout=500) as host:
yield host
def transformers_host(transformers_image):
with run_api_server_docker_container(transformers_image,
timeout=500) as host:
yield host
def tf2_host(tf2_image):
with run_api_server_docker_container(tf2_image, timeout=500) as host:
yield host
def h2o_docker_host(h2o_image):
with run_api_server_docker_container(h2o_image, timeout=500) as host:
yield host