5cfa1934e4a2157d07ef534e1033a3f0e5bc1082,python/ray/experimental/client/server/server.py,RayletServicer,_schedule_actor,#RayletServicer#Any#Any#Any#,181

Before Change

                        task: ray_client_pb2.ClientTask,
                        context=None,
                        prepared_args=None) -> ray_client_pb2.ClientTaskTicket:
        with stash_api_for_tests(self._test_mode):
            payload_ref = cloudpickle.loads(task.payload_id)
            if payload_ref.binary() not in self.registered_actor_classes:
                actor_class_ref = self.object_refs[payload_ref.binary()]
                actor_class = ray.get(actor_class_ref)
                if not inspect.isclass(actor_class):
                    raise Exception("Attempting to schedule actor that "
                                    "isn"t a class.")
                reg_class = ray.remote(actor_class)
                self.registered_actor_classes[payload_ref.binary()] = reg_class
            remote_class = self.registered_actor_classes[payload_ref.binary()]
            arglist = _convert_args(task.args, prepared_args)
            actor = remote_class.remote(*arglist)
            actorhandle = cloudpickle.dumps(actor)
            self.actor_refs[actorhandle] = actor
        return ray_client_pb2.ClientTaskTicket(
            return_ref=make_remote_ref(actor._actor_id.binary(), actorhandle))

    def _schedule_function(

After Change

        arglist = self._convert_args(task.args, prepared_args)
        actor = remote_class.remote(*arglist)
        self.actor_refs[actor._actor_id.binary()] = actor
        self.actor_owners[task.client_id].add(actor._actor_id.binary())
        return ray_client_pb2.ClientTaskTicket(
            return_id=actor._actor_id.binary())

Instances