[Placement group] Input validation (#11152)

* Add a basic input validation. * Addressed code review.
2026-06-27 22:08:16 +08:00 · 2020-10-14 13:56:41 -07:00
parent a10e36ca04
commit 666fcde8ca
2 changed files with 21 additions and 7 deletions
@@ -32,11 +32,17 @@ def test_placement_group_pack(ray_start_cluster):
    ray.init(address=cluster.address)

    placement_group = ray.util.placement_group(
-        name="name", strategy="PACK", bundles=[{
-            "CPU": 2
-        }, {
-            "CPU": 2
-        }])
+        name="name",
+        strategy="PACK",
+        bundles=[
+            {
+                "CPU": 2,
+                "GPU": 0  # Test 0 resource spec doesn't break tests.
+            },
+            {
+                "CPU": 2
+            }
+        ])
    ray.get(placement_group.ready())
    actor_1 = Actor.options(
        placement_group=placement_group,