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_image(transformers_svc_saved_dir):
with build_api_server_docker_image(transformers_svc_saved_dir,
"tranformers_example_service") as image:
yield image
def tf2_image(tf2_svc_saved_dir):
with build_api_server_docker_image(
tf2_svc_saved_dir, "tf2_example_service"
) as image:
yield image
def h2o_image(h2o_svc_saved_dir):
with build_api_server_docker_image(h2o_svc_saved_dir,
"h2o_example_service") as image:
yield image