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

[Placement Group]Add get all placement group api (#11460)

Browse Source 
* add get all interface for placement group

* add get all interface for placement group

* make it work

* fix lint

* fix lint

* fix comment

* add cpp test

* fix python lint
This commit is contained in:
DK.Pino
2020-10-24 02:46:48 +08:00
committed by GitHub
parent e7aa6441b7
commit 9f804ade5f
18 changed files with 157 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/util/placement_group.py
+4 -3
View File
@@ -185,17 +185,18 @@ def remove_placement_group(placement_group: PlacementGroup):
worker.core_worker.remove_placement_group(placement_group.id)
def placement_group_table(placement_group: PlacementGroup) -> dict:
def placement_group_table(placement_group: PlacementGroup = None) -> list:
"""Get the state of the placement group from GCS.
Args:
placement_group (PlacementGroup): placement group to see
states.
"""
assert placement_group is not None
worker = ray.worker.global_worker
worker.check_connected()
return ray.state.state.placement_group_table(placement_group.id)
placement_group_id = placement_group.id if (placement_group is
not None) else None
return ray.state.state.placement_group_table(placement_group_id)
def get_current_placement_group() -> Optional[PlacementGroup]: