From 429c0507cf7704f31021d66e2a1c64811932c7d9 Mon Sep 17 00:00:00 2001 From: SangBin Cho Date: Tue, 9 Jun 2020 10:17:18 -0700 Subject: [PATCH] [Serve] Serve long running test fix (#8864) --- ci/long_running_tests/workloads/serve.py | 3 +-- ci/long_running_tests/workloads/serve_failure.py | 4 ++-- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/ci/long_running_tests/workloads/serve.py b/ci/long_running_tests/workloads/serve.py index cb05af6d2..f3cf6a66e 100644 --- a/ci/long_running_tests/workloads/serve.py +++ b/ci/long_running_tests/workloads/serve.py @@ -42,10 +42,9 @@ def echo(_): return ["hi {}".format(i) for i in range(serve.context.batch_size)] -serve.create_endpoint("echo", "/echo") config = {"num_replicas": 30, "max_batch_size": 16} serve.create_backend("echo:v1", echo, config=config) -serve.set_traffic("echo", {"echo:v1": 1}) +serve.create_endpoint("echo", backend="echo:v1", route="/echo") print("Warming up") for _ in range(5): diff --git a/ci/long_running_tests/workloads/serve_failure.py b/ci/long_running_tests/workloads/serve_failure.py index 6166b55b8..9dc80acdb 100644 --- a/ci/long_running_tests/workloads/serve_failure.py +++ b/ci/long_running_tests/workloads/serve_failure.py @@ -80,8 +80,8 @@ class RandomTest: return new_endpoint serve.create_backend(new_endpoint, handler) - serve.create_endpoint(new_endpoint, "/" + new_endpoint) - serve.set_traffic(new_endpoint, {new_endpoint: 1.0}) + serve.create_endpoint( + new_endpoint, backend=new_endpoint, route="/" + new_endpoint) self.endpoints.append(new_endpoint)