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

[serve] Fix flaky test_api.py::test_backend_user_config (#12892)

Browse Source
This commit is contained in:
Edward Oakes
2020-12-15 15:35:30 -06:00
committed by GitHub
parent ea1228074d
commit 6795d7c75c
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/serve/tests/test_api.py
+6 -5
View File
@@ -112,18 +112,19 @@ def test_backend_user_config(serve_instance):
pids_seen = set()
for i in range(100):
result = ray.get(handle.remote())
assert (str(result[0]) == val), result[0]
if str(result[0]) != val:
return False
pids_seen.add(result[1])
assert (len(pids_seen) == num_replicas)
return len(pids_seen) == num_replicas
check("123", 2)
wait_for_condition(lambda: check("123", 2))
client.update_backend_config("counter", BackendConfig(num_replicas=3))
check("123", 3)
wait_for_condition(lambda: check("123", 3))
config = BackendConfig(user_config={"count": 456})
client.update_backend_config("counter", config)
check("456", 3)
wait_for_condition(lambda: check("456", 3))
def test_call_method(serve_instance):