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

[serve] Add serve.shutdown() (#8766)

Browse Source
This commit is contained in:
Edward Oakes
2020-06-23 13:42:03 -05:00
committed by GitHub
parent b6d425526d
commit c9010eb8ad
8 changed files with 77 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/experimental/internal_kv.py
+5 -5
View File
@@ -1,7 +1,5 @@
import ray
_local = {} # dict for local mode
def _internal_kv_initialized():
worker = ray.worker.global_worker
@@ -11,9 +9,7 @@ def _internal_kv_initialized():
def _internal_kv_get(key):
"""Fetch the value of a binary key."""
worker = ray.worker.global_worker
return worker.redis_client.hget(key, "value")
return ray.worker.global_worker.redis_client.hget(key, "value")
def _internal_kv_put(key, value, overwrite=False):
@@ -32,3 +28,7 @@ def _internal_kv_put(key, value, overwrite=False):
else:
updated = worker.redis_client.hsetnx(key, "value", value)
return updated == 0 # already exists
def _internal_kv_del(key):
return ray.worker.global_worker.redis_client.delete(key)