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

Revert "[PlacementGroup]Add PlacementGroup wait python api (#12601)" (#12825)

Browse Source 
This reverts commit 401d342602.
This commit is contained in:
Eric Liang
2020-12-12 12:13:48 -08:00
committed by GitHub
parent b73d4831d4
commit bdc6624da8
2 changed files with 7 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tests/test_placement_group.py
+7 -21
View File
@@ -50,7 +50,7 @@ def test_placement_group_pack(ray_start_cluster):
"CPU": 2
}
])
assert placement_group.wait(10000)
ray.get(placement_group.ready())
actor_1 = Actor.options(
placement_group=placement_group,
placement_group_bundle_index=0).remote()
@@ -96,7 +96,7 @@ def test_placement_group_strict_pack(ray_start_cluster):
}, {
"CPU": 2
}])
assert placement_group.wait(10000)
ray.get(placement_group.ready())
actor_1 = Actor.options(
placement_group=placement_group,
placement_group_bundle_index=0).remote()
@@ -142,7 +142,7 @@ def test_placement_group_spread(ray_start_cluster):
}, {
"CPU": 2
}])
assert placement_group.wait(10000)
ray.get(placement_group.ready())
actor_1 = Actor.options(
placement_group=placement_group,
placement_group_bundle_index=0).remote()
@@ -192,7 +192,7 @@ def test_placement_group_strict_spread(ray_start_cluster):
}, {
"CPU": 2
}])
assert placement_group.wait(10000)
ray.get(placement_group.ready())
actor_1 = Actor.options(
placement_group=placement_group,
placement_group_bundle_index=0).remote()
@@ -1177,7 +1177,7 @@ def test_create_placement_group_after_gcs_server_restart(
# Create placement group 1 successfully.
placement_group1 = ray.util.placement_group([{"CPU": 1}, {"CPU": 1}])
assert placement_group1.wait(10000)
ray.get(placement_group1.ready(), timeout=10)
table = ray.util.placement_group_table(placement_group1)
assert table["state"] == "CREATED"
@@ -1187,7 +1187,7 @@ def test_create_placement_group_after_gcs_server_restart(
# Create placement group 2 successfully.
placement_group2 = ray.util.placement_group([{"CPU": 1}, {"CPU": 1}])
assert placement_group2.wait(10000)
ray.get(placement_group2.ready(), timeout=10)
table = ray.util.placement_group_table(placement_group2)
assert table["state"] == "CREATED"
@@ -1250,21 +1250,7 @@ def test_create_placement_group_during_gcs_server_restart(
cluster.head_node.start_gcs_server()
for i in range(0, 100):
assert placement_groups[i].wait(10000)
def test_placement_group_wait_api(ray_start_cluster):
cluster = ray_start_cluster
cluster.add_node(num_cpus=4)
ray.init(address=cluster.address)
placement_group = ray.util.placement_group(
name="name", strategy="PACK", bundles=[{
"CPU": 2,
}, {
"CPU": 2
}])
assert placement_group.wait(10000)
ray.get(placement_groups[i].ready())
if __name__ == "__main__":