wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-13 17:45:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

[serve] Serve client refactor (#10409)

Browse Source
This commit is contained in:
Edward Oakes
2020-09-04 12:02:23 -05:00
committed by GitHub
parent 2e49e22f21
commit 786f12edfd
45 changed files with 1017 additions and 929 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/serve/tests/test_handle.py
+7 -6
View File
@@ -5,7 +5,7 @@ import requests
def test_handle_in_endpoint(serve_instance):
serve.init()
client = serve_instance
class Endpoint1:
def __call__(self, flask_request):
@@ -13,20 +13,21 @@ def test_handle_in_endpoint(serve_instance):
class Endpoint2:
def __init__(self):
self.handle = serve.get_handle("endpoint1", missing_ok=True)
client = serve.connect()
self.handle = client.get_handle("endpoint1")
def __call__(self):
return ray.get(self.handle.remote())
serve.create_backend("endpoint1:v0", Endpoint1)
serve.create_endpoint(
client.create_backend("endpoint1:v0", Endpoint1)
client.create_endpoint(
"endpoint1",
backend="endpoint1:v0",
route="/endpoint1",
methods=["GET", "POST"])
serve.create_backend("endpoint2:v0", Endpoint2)
serve.create_endpoint(
client.create_backend("endpoint2:v0", Endpoint2)
client.create_endpoint(
"endpoint2",
backend="endpoint2:v0",
route="/endpoint2",