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

Add gcs_service_enabled function to avoid getting environment variable directly (#7742)

Browse Source
This commit is contained in:
fangfengbin
2020-03-26 22:02:53 +08:00
committed by GitHub
parent ca6eabc9cb
commit e196fcdbaf
12 changed files with 31 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tests/test_component_failures_3.py
+2 -2
View File
@@ -82,7 +82,7 @@ def test_driver_lives_sequential(ray_start_regular):
ray.worker._global_node.kill_plasma_store()
ray.worker._global_node.kill_log_monitor()
ray.worker._global_node.kill_monitor()
if os.environ.get(ray_constants.RAY_GCS_SERVICE_ENABLED, True):
if ray_constants.GCS_SERVICE_ENABLED:
ray.worker._global_node.kill_gcs_server()
else:
ray.worker._global_node.kill_raylet_monitor()
@@ -96,7 +96,7 @@ def test_driver_lives_sequential(ray_start_regular):
def test_driver_lives_parallel(ray_start_regular):
all_processes = ray.worker._global_node.all_processes
if os.environ.get(ray_constants.RAY_GCS_SERVICE_ENABLED, True):
if ray_constants.GCS_SERVICE_ENABLED:
process_infos = (all_processes[ray_constants.PROCESS_TYPE_PLASMA_STORE]
+ all_processes[ray_constants.PROCESS_TYPE_GCS_SERVER]
+ all_processes[ray_constants.PROCESS_TYPE_RAYLET] +