5cfa1934e4a2157d07ef534e1033a3f0e5bc1082,python/ray/experimental/client/server/server.py,RayletServicer,WaitObject,#RayletServicer#Any#Any#,112

Before Change

            ) for ready_object_ref in ready_object_refs
        ]
        remaining_object_ids = [
            make_remote_ref(
                id=remaining_object_ref.binary(),
                handle=cloudpickle.dumps(remaining_object_ref),
            ) for remaining_object_ref in remaining_object_refs
        ]
        return ray_client_pb2.WaitResponse(
            valid=True,

After Change

    def WaitObject(self, request, context=None) -> ray_client_pb2.WaitResponse:
        object_refs = []
        for id in request.object_ids:
            if id not in self.object_refs[request.client_id]:
                raise Exception(
                    "Asking for a ref not associated with this client: %s" %
                    str(id))

Instances