1e39c403709b96d0bfa9c139e927441acc232426,python/ray/tests/test_placement_group.py,,test_capture_child_tasks,#Any#,796

Before Change

            ray.get(actor.ready.remote())
            self.actors.append(actor)

    a = Actor.options(placement_group=pg).remote()
    ray.get(a.ready.remote())
    // 1 top level actor + 3 children.
    for _ in range(total_num_actors - 1):
        ray.get(a.schedule_nested_actor.remote())
    // Make sure all the actors are scheduled on the same node.
    // (why? The placement group has STRICT_PACK strategy).
    node_id_set = set()

After Change

            for _ in range(3)
        ])

    t = create_nested_task.options(
        num_cpus=1, num_gpus=0, placement_group=pg).remote(1, 0)
    pgs = ray.get(t)
    // Every task should have current placement group because they
    // should be implicitly captured by default.
    assert None not in pgs

Instances