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

Add delete_creating_tasks option for internal.free() (#4588)

Browse Source 
* add delete creating task objects.

* format code style

* Fix lint

* add tests add address comments.

* Refine test

* Refine java test

* Fix CI

* Refine

* Fix lint

* Fix CI
This commit is contained in:
Wang Qing
2019-04-12 13:38:31 +08:00
committed by Yuhong Guo
parent e88e706fcc
commit fe07a5b4b1
19 changed files with 131 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/internal/internal_api.py
+5 -2
View File
@@ -8,7 +8,7 @@ from ray import profiling
__all__ = ["free"]
def free(object_ids, local_only=False):
def free(object_ids, local_only=False, delete_creating_tasks=False):
"""Free a list of IDs from object stores.
This function is a low-level API which should be used in restricted
@@ -25,6 +25,8 @@ def free(object_ids, local_only=False):
object_ids (List[ObjectID]): List of object IDs to delete.
local_only (bool): Whether only deleting the list of objects in local
object store or all object stores.
delete_creating_tasks (bool): Whether also delete the object creating
tasks.
"""
worker = ray.worker.get_global_worker()
@@ -46,4 +48,5 @@ def free(object_ids, local_only=False):
if len(object_ids) == 0:
return
worker.raylet_client.free_objects(object_ids, local_only)
worker.raylet_client.free_objects(object_ids, local_only,
delete_creating_tasks)