hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 10:31:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix core worker shutdown errors (#6004)

Browse source
This commit is contained in:
Edward Oakes 2019-10-24 22:29:05 -07:00 • committed by GitHub
parent 71a2f4c63d
commit 6f27d881bd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
python/ray/worker.py
View file

@ -1669,12 +1669,11 @@ def disconnect(exiting_interpreter=False):
worker.function_actor_manager.reset_cache() worker.function_actor_manager.reset_cache()
worker.serialization_context_map.clear() worker.serialization_context_map.clear()
if not exiting_interpreter: # We need to destruct the core worker here because after this function,
if hasattr(worker, "raylet_client"): # we will tear down any processes spawned by ray.init() and the background
del worker.raylet_client # threads in the core worker don't currently handle that gracefully.
if hasattr(worker, "core_worker"):
if hasattr(worker, "core_worker"): del worker.core_worker
del worker.core_worker
@contextmanager @contextmanager