Add files using upload-large-folder tool

.venv/lib/python3.11/site-packages/ray/core/generated/core_worker_pb2_grpc.py

@@ -0,0 +1,826 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+
+from . import core_worker_pb2 as src_dot_ray_dot_protobuf_dot_core__worker__pb2
+from . import pubsub_pb2 as src_dot_ray_dot_protobuf_dot_pubsub__pb2
+
+
+class CoreWorkerServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.RayletNotifyGCSRestart = channel.unary_unary(
+                '/ray.rpc.CoreWorkerService/RayletNotifyGCSRestart',
+                request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RayletNotifyGCSRestartRequest.SerializeToString,
+                response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RayletNotifyGCSRestartReply.FromString,
+                )
+        self.PushTask = channel.unary_unary(
+                '/ray.rpc.CoreWorkerService/PushTask',
+                request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.PushTaskRequest.SerializeToString,
+                response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.PushTaskReply.FromString,
+                )
+        self.DirectActorCallArgWaitComplete = channel.unary_unary(
+                '/ray.rpc.CoreWorkerService/DirectActorCallArgWaitComplete',
+                request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.DirectActorCallArgWaitCompleteRequest.SerializeToString,
+                response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.DirectActorCallArgWaitCompleteReply.FromString,
+                )
+        self.GetObjectStatus = channel.unary_unary(
+                '/ray.rpc.CoreWorkerService/GetObjectStatus',
+                request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetObjectStatusRequest.SerializeToString,
+                response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetObjectStatusReply.FromString,
+                )
+        self.WaitForActorRefDeleted = channel.unary_unary(
+                '/ray.rpc.CoreWorkerService/WaitForActorRefDeleted',
+                request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.WaitForActorRefDeletedRequest.SerializeToString,
+                response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.WaitForActorRefDeletedReply.FromString,
+                )
+        self.PubsubLongPolling = channel.unary_unary(
+                '/ray.rpc.CoreWorkerService/PubsubLongPolling',
+                request_serializer=src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubLongPollingRequest.SerializeToString,
+                response_deserializer=src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubLongPollingReply.FromString,
+                )
+        self.ReportGeneratorItemReturns = channel.unary_unary(
+                '/ray.rpc.CoreWorkerService/ReportGeneratorItemReturns',
+                request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.ReportGeneratorItemReturnsRequest.SerializeToString,
+                response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.ReportGeneratorItemReturnsReply.FromString,
+                )
+        self.PubsubCommandBatch = channel.unary_unary(
+                '/ray.rpc.CoreWorkerService/PubsubCommandBatch',
+                request_serializer=src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubCommandBatchRequest.SerializeToString,
+                response_deserializer=src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubCommandBatchReply.FromString,
+                )
+        self.UpdateObjectLocationBatch = channel.unary_unary(
+                '/ray.rpc.CoreWorkerService/UpdateObjectLocationBatch',
+                request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.UpdateObjectLocationBatchRequest.SerializeToString,
+                response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.UpdateObjectLocationBatchReply.FromString,
+                )
+        self.GetObjectLocationsOwner = channel.unary_unary(
+                '/ray.rpc.CoreWorkerService/GetObjectLocationsOwner',
+                request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetObjectLocationsOwnerRequest.SerializeToString,
+                response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetObjectLocationsOwnerReply.FromString,
+                )
+        self.KillActor = channel.unary_unary(
+                '/ray.rpc.CoreWorkerService/KillActor',
+                request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.KillActorRequest.SerializeToString,
+                response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.KillActorReply.FromString,
+                )
+        self.CancelTask = channel.unary_unary(
+                '/ray.rpc.CoreWorkerService/CancelTask',
+                request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.CancelTaskRequest.SerializeToString,
+                response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.CancelTaskReply.FromString,
+                )
+        self.RemoteCancelTask = channel.unary_unary(
+                '/ray.rpc.CoreWorkerService/RemoteCancelTask',
+                request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RemoteCancelTaskRequest.SerializeToString,
+                response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RemoteCancelTaskReply.FromString,
+                )
+        self.GetCoreWorkerStats = channel.unary_unary(
+                '/ray.rpc.CoreWorkerService/GetCoreWorkerStats',
+                request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetCoreWorkerStatsRequest.SerializeToString,
+                response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetCoreWorkerStatsReply.FromString,
+                )
+        self.LocalGC = channel.unary_unary(
+                '/ray.rpc.CoreWorkerService/LocalGC',
+                request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.LocalGCRequest.SerializeToString,
+                response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.LocalGCReply.FromString,
+                )
+        self.DeleteObjects = channel.unary_unary(
+                '/ray.rpc.CoreWorkerService/DeleteObjects',
+                request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.DeleteObjectsRequest.SerializeToString,
+                response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.DeleteObjectsReply.FromString,
+                )
+        self.SpillObjects = channel.unary_unary(
+                '/ray.rpc.CoreWorkerService/SpillObjects',
+                request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.SpillObjectsRequest.SerializeToString,
+                response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.SpillObjectsReply.FromString,
+                )
+        self.RestoreSpilledObjects = channel.unary_unary(
+                '/ray.rpc.CoreWorkerService/RestoreSpilledObjects',
+                request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RestoreSpilledObjectsRequest.SerializeToString,
+                response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RestoreSpilledObjectsReply.FromString,
+                )
+        self.DeleteSpilledObjects = channel.unary_unary(
+                '/ray.rpc.CoreWorkerService/DeleteSpilledObjects',
+                request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.DeleteSpilledObjectsRequest.SerializeToString,
+                response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.DeleteSpilledObjectsReply.FromString,
+                )
+        self.PlasmaObjectReady = channel.unary_unary(
+                '/ray.rpc.CoreWorkerService/PlasmaObjectReady',
+                request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.PlasmaObjectReadyRequest.SerializeToString,
+                response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.PlasmaObjectReadyReply.FromString,
+                )
+        self.Exit = channel.unary_unary(
+                '/ray.rpc.CoreWorkerService/Exit',
+                request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.ExitRequest.SerializeToString,
+                response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.ExitReply.FromString,
+                )
+        self.AssignObjectOwner = channel.unary_unary(
+                '/ray.rpc.CoreWorkerService/AssignObjectOwner',
+                request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.AssignObjectOwnerRequest.SerializeToString,
+                response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.AssignObjectOwnerReply.FromString,
+                )
+        self.NumPendingTasks = channel.unary_unary(
+                '/ray.rpc.CoreWorkerService/NumPendingTasks',
+                request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.NumPendingTasksRequest.SerializeToString,
+                response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.NumPendingTasksReply.FromString,
+                )
+        self.RegisterMutableObjectReader = channel.unary_unary(
+                '/ray.rpc.CoreWorkerService/RegisterMutableObjectReader',
+                request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RegisterMutableObjectReaderRequest.SerializeToString,
+                response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RegisterMutableObjectReaderReply.FromString,
+                )
+
+
+class CoreWorkerServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def RayletNotifyGCSRestart(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def PushTask(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def DirectActorCallArgWaitComplete(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetObjectStatus(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def WaitForActorRefDeleted(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def PubsubLongPolling(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def ReportGeneratorItemReturns(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def PubsubCommandBatch(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def UpdateObjectLocationBatch(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetObjectLocationsOwner(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def KillActor(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def CancelTask(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def RemoteCancelTask(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetCoreWorkerStats(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def LocalGC(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def DeleteObjects(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def SpillObjects(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def RestoreSpilledObjects(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def DeleteSpilledObjects(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def PlasmaObjectReady(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def Exit(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def AssignObjectOwner(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def NumPendingTasks(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def RegisterMutableObjectReader(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_CoreWorkerServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'RayletNotifyGCSRestart': grpc.unary_unary_rpc_method_handler(
+                    servicer.RayletNotifyGCSRestart,
+                    request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RayletNotifyGCSRestartRequest.FromString,
+                    response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RayletNotifyGCSRestartReply.SerializeToString,
+            ),
+            'PushTask': grpc.unary_unary_rpc_method_handler(
+                    servicer.PushTask,
+                    request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.PushTaskRequest.FromString,
+                    response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.PushTaskReply.SerializeToString,
+            ),
+            'DirectActorCallArgWaitComplete': grpc.unary_unary_rpc_method_handler(
+                    servicer.DirectActorCallArgWaitComplete,
+                    request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.DirectActorCallArgWaitCompleteRequest.FromString,
+                    response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.DirectActorCallArgWaitCompleteReply.SerializeToString,
+            ),
+            'GetObjectStatus': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetObjectStatus,
+                    request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetObjectStatusRequest.FromString,
+                    response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetObjectStatusReply.SerializeToString,
+            ),
+            'WaitForActorRefDeleted': grpc.unary_unary_rpc_method_handler(
+                    servicer.WaitForActorRefDeleted,
+                    request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.WaitForActorRefDeletedRequest.FromString,
+                    response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.WaitForActorRefDeletedReply.SerializeToString,
+            ),
+            'PubsubLongPolling': grpc.unary_unary_rpc_method_handler(
+                    servicer.PubsubLongPolling,
+                    request_deserializer=src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubLongPollingRequest.FromString,
+                    response_serializer=src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubLongPollingReply.SerializeToString,
+            ),
+            'ReportGeneratorItemReturns': grpc.unary_unary_rpc_method_handler(
+                    servicer.ReportGeneratorItemReturns,
+                    request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.ReportGeneratorItemReturnsRequest.FromString,
+                    response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.ReportGeneratorItemReturnsReply.SerializeToString,
+            ),
+            'PubsubCommandBatch': grpc.unary_unary_rpc_method_handler(
+                    servicer.PubsubCommandBatch,
+                    request_deserializer=src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubCommandBatchRequest.FromString,
+                    response_serializer=src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubCommandBatchReply.SerializeToString,
+            ),
+            'UpdateObjectLocationBatch': grpc.unary_unary_rpc_method_handler(
+                    servicer.UpdateObjectLocationBatch,
+                    request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.UpdateObjectLocationBatchRequest.FromString,
+                    response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.UpdateObjectLocationBatchReply.SerializeToString,
+            ),
+            'GetObjectLocationsOwner': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetObjectLocationsOwner,
+                    request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetObjectLocationsOwnerRequest.FromString,
+                    response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetObjectLocationsOwnerReply.SerializeToString,
+            ),
+            'KillActor': grpc.unary_unary_rpc_method_handler(
+                    servicer.KillActor,
+                    request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.KillActorRequest.FromString,
+                    response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.KillActorReply.SerializeToString,
+            ),
+            'CancelTask': grpc.unary_unary_rpc_method_handler(
+                    servicer.CancelTask,
+                    request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.CancelTaskRequest.FromString,
+                    response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.CancelTaskReply.SerializeToString,
+            ),
+            'RemoteCancelTask': grpc.unary_unary_rpc_method_handler(
+                    servicer.RemoteCancelTask,
+                    request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RemoteCancelTaskRequest.FromString,
+                    response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RemoteCancelTaskReply.SerializeToString,
+            ),
+            'GetCoreWorkerStats': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetCoreWorkerStats,
+                    request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetCoreWorkerStatsRequest.FromString,
+                    response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetCoreWorkerStatsReply.SerializeToString,
+            ),
+            'LocalGC': grpc.unary_unary_rpc_method_handler(
+                    servicer.LocalGC,
+                    request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.LocalGCRequest.FromString,
+                    response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.LocalGCReply.SerializeToString,
+            ),
+            'DeleteObjects': grpc.unary_unary_rpc_method_handler(
+                    servicer.DeleteObjects,
+                    request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.DeleteObjectsRequest.FromString,
+                    response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.DeleteObjectsReply.SerializeToString,
+            ),
+            'SpillObjects': grpc.unary_unary_rpc_method_handler(
+                    servicer.SpillObjects,
+                    request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.SpillObjectsRequest.FromString,
+                    response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.SpillObjectsReply.SerializeToString,
+            ),
+            'RestoreSpilledObjects': grpc.unary_unary_rpc_method_handler(
+                    servicer.RestoreSpilledObjects,
+                    request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RestoreSpilledObjectsRequest.FromString,
+                    response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RestoreSpilledObjectsReply.SerializeToString,
+            ),
+            'DeleteSpilledObjects': grpc.unary_unary_rpc_method_handler(
+                    servicer.DeleteSpilledObjects,
+                    request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.DeleteSpilledObjectsRequest.FromString,
+                    response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.DeleteSpilledObjectsReply.SerializeToString,
+            ),
+            'PlasmaObjectReady': grpc.unary_unary_rpc_method_handler(
+                    servicer.PlasmaObjectReady,
+                    request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.PlasmaObjectReadyRequest.FromString,
+                    response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.PlasmaObjectReadyReply.SerializeToString,
+            ),
+            'Exit': grpc.unary_unary_rpc_method_handler(
+                    servicer.Exit,
+                    request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.ExitRequest.FromString,
+                    response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.ExitReply.SerializeToString,
+            ),
+            'AssignObjectOwner': grpc.unary_unary_rpc_method_handler(
+                    servicer.AssignObjectOwner,
+                    request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.AssignObjectOwnerRequest.FromString,
+                    response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.AssignObjectOwnerReply.SerializeToString,
+            ),
+            'NumPendingTasks': grpc.unary_unary_rpc_method_handler(
+                    servicer.NumPendingTasks,
+                    request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.NumPendingTasksRequest.FromString,
+                    response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.NumPendingTasksReply.SerializeToString,
+            ),
+            'RegisterMutableObjectReader': grpc.unary_unary_rpc_method_handler(
+                    servicer.RegisterMutableObjectReader,
+                    request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RegisterMutableObjectReaderRequest.FromString,
+                    response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RegisterMutableObjectReaderReply.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'ray.rpc.CoreWorkerService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class CoreWorkerService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def RayletNotifyGCSRestart(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/RayletNotifyGCSRestart',
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.RayletNotifyGCSRestartRequest.SerializeToString,
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.RayletNotifyGCSRestartReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def PushTask(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/PushTask',
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.PushTaskRequest.SerializeToString,
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.PushTaskReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def DirectActorCallArgWaitComplete(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/DirectActorCallArgWaitComplete',
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.DirectActorCallArgWaitCompleteRequest.SerializeToString,
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.DirectActorCallArgWaitCompleteReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetObjectStatus(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/GetObjectStatus',
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetObjectStatusRequest.SerializeToString,
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetObjectStatusReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def WaitForActorRefDeleted(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/WaitForActorRefDeleted',
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.WaitForActorRefDeletedRequest.SerializeToString,
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.WaitForActorRefDeletedReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def PubsubLongPolling(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/PubsubLongPolling',
+            src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubLongPollingRequest.SerializeToString,
+            src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubLongPollingReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def ReportGeneratorItemReturns(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/ReportGeneratorItemReturns',
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.ReportGeneratorItemReturnsRequest.SerializeToString,
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.ReportGeneratorItemReturnsReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def PubsubCommandBatch(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/PubsubCommandBatch',
+            src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubCommandBatchRequest.SerializeToString,
+            src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubCommandBatchReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def UpdateObjectLocationBatch(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/UpdateObjectLocationBatch',
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.UpdateObjectLocationBatchRequest.SerializeToString,
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.UpdateObjectLocationBatchReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetObjectLocationsOwner(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/GetObjectLocationsOwner',
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetObjectLocationsOwnerRequest.SerializeToString,
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetObjectLocationsOwnerReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def KillActor(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/KillActor',
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.KillActorRequest.SerializeToString,
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.KillActorReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def CancelTask(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/CancelTask',
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.CancelTaskRequest.SerializeToString,
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.CancelTaskReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def RemoteCancelTask(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/RemoteCancelTask',
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.RemoteCancelTaskRequest.SerializeToString,
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.RemoteCancelTaskReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetCoreWorkerStats(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/GetCoreWorkerStats',
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetCoreWorkerStatsRequest.SerializeToString,
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetCoreWorkerStatsReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def LocalGC(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/LocalGC',
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.LocalGCRequest.SerializeToString,
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.LocalGCReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def DeleteObjects(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/DeleteObjects',
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.DeleteObjectsRequest.SerializeToString,
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.DeleteObjectsReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def SpillObjects(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/SpillObjects',
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.SpillObjectsRequest.SerializeToString,
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.SpillObjectsReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def RestoreSpilledObjects(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/RestoreSpilledObjects',
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.RestoreSpilledObjectsRequest.SerializeToString,
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.RestoreSpilledObjectsReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def DeleteSpilledObjects(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/DeleteSpilledObjects',
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.DeleteSpilledObjectsRequest.SerializeToString,
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.DeleteSpilledObjectsReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def PlasmaObjectReady(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/PlasmaObjectReady',
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.PlasmaObjectReadyRequest.SerializeToString,
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.PlasmaObjectReadyReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def Exit(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/Exit',
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.ExitRequest.SerializeToString,
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.ExitReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def AssignObjectOwner(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/AssignObjectOwner',
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.AssignObjectOwnerRequest.SerializeToString,
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.AssignObjectOwnerReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def NumPendingTasks(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/NumPendingTasks',
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.NumPendingTasksRequest.SerializeToString,
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.NumPendingTasksReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def RegisterMutableObjectReader(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/RegisterMutableObjectReader',
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.RegisterMutableObjectReaderRequest.SerializeToString,
+            src_dot_ray_dot_protobuf_dot_core__worker__pb2.RegisterMutableObjectReaderReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

.venv/lib/python3.11/site-packages/ray/core/generated/export_actor_data_pb2.py

@@ -0,0 +1,51 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: src/ray/protobuf/export_api/export_actor_data.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from . import common_pb2 as src_dot_ray_dot_protobuf_dot_common__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n3src/ray/protobuf/export_api/export_actor_data.proto\x12\x07ray.rpc\x1a\x1dsrc/ray/protobuf/common.proto\"\x94\x06\n\x0f\x45xportActorData\x12\x19\n\x08\x61\x63tor_id\x18\x01 \x01(\x0cR\x07\x61\x63torId\x12\x15\n\x06job_id\x18\x02 \x01(\x0cR\x05jobId\x12\x39\n\x05state\x18\x03 \x01(\x0e\x32#.ray.rpc.ExportActorData.ActorStateR\x05state\x12\x1f\n\x0bis_detached\x18\x04 \x01(\x08R\nisDetached\x12\x12\n\x04name\x18\x05 \x01(\tR\x04name\x12\x10\n\x03pid\x18\x06 \x01(\rR\x03pid\x12#\n\rray_namespace\x18\x07 \x01(\tR\x0crayNamespace\x12\x34\n\x16serialized_runtime_env\x18\x08 \x01(\tR\x14serializedRuntimeEnv\x12\x1d\n\nclass_name\x18\t \x01(\tR\tclassName\x12\x39\n\x0b\x64\x65\x61th_cause\x18\n \x01(\x0b\x32\x18.ray.rpc.ActorDeathCauseR\ndeathCause\x12^\n\x12required_resources\x18\x0b \x03(\x0b\x32/.ray.rpc.ExportActorData.RequiredResourcesEntryR\x11requiredResources\x12\x1c\n\x07node_id\x18\x0c \x01(\x0cH\x00R\x06nodeId\x88\x01\x01\x12\x31\n\x12placement_group_id\x18\r \x01(\x0cH\x01R\x10placementGroupId\x88\x01\x01\x12\x1b\n\trepr_name\x18\x0e \x01(\tR\x08reprName\x1a\x44\n\x16RequiredResourcesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\x01R\x05value:\x02\x38\x01\"a\n\nActorState\x12\x18\n\x14\x44\x45PENDENCIES_UNREADY\x10\x00\x12\x14\n\x10PENDING_CREATION\x10\x01\x12\t\n\x05\x41LIVE\x10\x02\x12\x0e\n\nRESTARTING\x10\x03\x12\x08\n\x04\x44\x45\x41\x44\x10\x04\x42\n\n\x08_node_idB\x15\n\x13_placement_group_idb\x06proto3')
+
+
+
+_EXPORTACTORDATA = DESCRIPTOR.message_types_by_name['ExportActorData']
+_EXPORTACTORDATA_REQUIREDRESOURCESENTRY = _EXPORTACTORDATA.nested_types_by_name['RequiredResourcesEntry']
+_EXPORTACTORDATA_ACTORSTATE = _EXPORTACTORDATA.enum_types_by_name['ActorState']
+ExportActorData = _reflection.GeneratedProtocolMessageType('ExportActorData', (_message.Message,), {
+
+  'RequiredResourcesEntry' : _reflection.GeneratedProtocolMessageType('RequiredResourcesEntry', (_message.Message,), {
+    'DESCRIPTOR' : _EXPORTACTORDATA_REQUIREDRESOURCESENTRY,
+    '__module__' : 'src.ray.protobuf.export_api.export_actor_data_pb2'
+    # @@protoc_insertion_point(class_scope:ray.rpc.ExportActorData.RequiredResourcesEntry)
+    })
+  ,
+  'DESCRIPTOR' : _EXPORTACTORDATA,
+  '__module__' : 'src.ray.protobuf.export_api.export_actor_data_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.ExportActorData)
+  })
+_sym_db.RegisterMessage(ExportActorData)
+_sym_db.RegisterMessage(ExportActorData.RequiredResourcesEntry)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  _EXPORTACTORDATA_REQUIREDRESOURCESENTRY._options = None
+  _EXPORTACTORDATA_REQUIREDRESOURCESENTRY._serialized_options = b'8\001'
+  _EXPORTACTORDATA._serialized_start=96
+  _EXPORTACTORDATA._serialized_end=884
+  _EXPORTACTORDATA_REQUIREDRESOURCESENTRY._serialized_start=682
+  _EXPORTACTORDATA_REQUIREDRESOURCESENTRY._serialized_end=750
+  _EXPORTACTORDATA_ACTORSTATE._serialized_start=752
+  _EXPORTACTORDATA_ACTORSTATE._serialized_end=849
+# @@protoc_insertion_point(module_scope)

.venv/lib/python3.11/site-packages/ray/core/generated/export_runtime_env_pb2.py

@@ -0,0 +1,56 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: src/ray/protobuf/export_api/export_runtime_env.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n4src/ray/protobuf/export_api/export_runtime_env.proto\x12\x07ray.rpc\"\xd9\x03\n\x14\x45xportRuntimeEnvInfo\x12\x34\n\x16serialized_runtime_env\x18\x01 \x01(\tR\x14serializedRuntimeEnv\x12@\n\x04uris\x18\x02 \x01(\x0b\x32,.ray.rpc.ExportRuntimeEnvInfo.RuntimeEnvUrisR\x04uris\x12\\\n\x12runtime_env_config\x18\x03 \x01(\x0b\x32..ray.rpc.ExportRuntimeEnvInfo.RuntimeEnvConfigR\x10runtimeEnvConfig\x1a`\n\x0eRuntimeEnvUris\x12&\n\x0fworking_dir_uri\x18\x01 \x01(\tR\rworkingDirUri\x12&\n\x0fpy_modules_uris\x18\x02 \x03(\tR\rpyModulesUris\x1a\x88\x01\n\x10RuntimeEnvConfig\x12\x32\n\x15setup_timeout_seconds\x18\x01 \x01(\x05R\x13setupTimeoutSeconds\x12#\n\reager_install\x18\x02 \x01(\x08R\x0c\x65\x61gerInstall\x12\x1b\n\tlog_files\x18\x03 \x03(\tR\x08logFilesb\x06proto3')
+
+
+
+_EXPORTRUNTIMEENVINFO = DESCRIPTOR.message_types_by_name['ExportRuntimeEnvInfo']
+_EXPORTRUNTIMEENVINFO_RUNTIMEENVURIS = _EXPORTRUNTIMEENVINFO.nested_types_by_name['RuntimeEnvUris']
+_EXPORTRUNTIMEENVINFO_RUNTIMEENVCONFIG = _EXPORTRUNTIMEENVINFO.nested_types_by_name['RuntimeEnvConfig']
+ExportRuntimeEnvInfo = _reflection.GeneratedProtocolMessageType('ExportRuntimeEnvInfo', (_message.Message,), {
+
+  'RuntimeEnvUris' : _reflection.GeneratedProtocolMessageType('RuntimeEnvUris', (_message.Message,), {
+    'DESCRIPTOR' : _EXPORTRUNTIMEENVINFO_RUNTIMEENVURIS,
+    '__module__' : 'src.ray.protobuf.export_api.export_runtime_env_pb2'
+    # @@protoc_insertion_point(class_scope:ray.rpc.ExportRuntimeEnvInfo.RuntimeEnvUris)
+    })
+  ,
+
+  'RuntimeEnvConfig' : _reflection.GeneratedProtocolMessageType('RuntimeEnvConfig', (_message.Message,), {
+    'DESCRIPTOR' : _EXPORTRUNTIMEENVINFO_RUNTIMEENVCONFIG,
+    '__module__' : 'src.ray.protobuf.export_api.export_runtime_env_pb2'
+    # @@protoc_insertion_point(class_scope:ray.rpc.ExportRuntimeEnvInfo.RuntimeEnvConfig)
+    })
+  ,
+  'DESCRIPTOR' : _EXPORTRUNTIMEENVINFO,
+  '__module__' : 'src.ray.protobuf.export_api.export_runtime_env_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.ExportRuntimeEnvInfo)
+  })
+_sym_db.RegisterMessage(ExportRuntimeEnvInfo)
+_sym_db.RegisterMessage(ExportRuntimeEnvInfo.RuntimeEnvUris)
+_sym_db.RegisterMessage(ExportRuntimeEnvInfo.RuntimeEnvConfig)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  _EXPORTRUNTIMEENVINFO._serialized_start=66
+  _EXPORTRUNTIMEENVINFO._serialized_end=539
+  _EXPORTRUNTIMEENVINFO_RUNTIMEENVURIS._serialized_start=304
+  _EXPORTRUNTIMEENVINFO_RUNTIMEENVURIS._serialized_end=400
+  _EXPORTRUNTIMEENVINFO_RUNTIMEENVCONFIG._serialized_start=403
+  _EXPORTRUNTIMEENVINFO_RUNTIMEENVCONFIG._serialized_end=539
+# @@protoc_insertion_point(module_scope)

.venv/lib/python3.11/site-packages/ray/core/generated/logging_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

.venv/lib/python3.11/site-packages/ray/core/generated/node_manager_pb2.py

@@ -0,0 +1,534 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: src/ray/protobuf/node_manager.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from . import common_pb2 as src_dot_ray_dot_protobuf_dot_common__pb2
+from . import gcs_pb2 as src_dot_ray_dot_protobuf_dot_gcs__pb2
+from . import autoscaler_pb2 as src_dot_ray_dot_protobuf_dot_autoscaler__pb2
+from . import runtime_env_common_pb2 as src_dot_ray_dot_protobuf_dot_runtime__env__common__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n#src/ray/protobuf/node_manager.proto\x12\x07ray.rpc\x1a\x1dsrc/ray/protobuf/common.proto\x1a\x1asrc/ray/protobuf/gcs.proto\x1a!src/ray/protobuf/autoscaler.proto\x1a)src/ray/protobuf/runtime_env_common.proto\"p\n\x13WorkerBacklogReport\x12\x36\n\rresource_spec\x18\x01 \x01(\x0b\x32\x11.ray.rpc.TaskSpecR\x0cresourceSpec\x12!\n\x0c\x62\x61\x63klog_size\x18\x02 \x01(\x03R\x0b\x62\x61\x63klogSize\"\x80\x01\n\x1aReportWorkerBacklogRequest\x12\x1b\n\tworker_id\x18\x01 \x01(\x0cR\x08workerId\x12\x45\n\x0f\x62\x61\x63klog_reports\x18\x02 \x03(\x0b\x32\x1c.ray.rpc.WorkerBacklogReportR\x0e\x62\x61\x63klogReports\"\x1a\n\x18ReportWorkerBacklogReply\"\xe0\x01\n\x19RequestWorkerLeaseRequest\x12\x36\n\rresource_spec\x18\x01 \x01(\x0b\x32\x11.ray.rpc.TaskSpecR\x0cresourceSpec\x12!\n\x0c\x62\x61\x63klog_size\x18\x02 \x01(\x03R\x0b\x62\x61\x63klogSize\x12&\n\x0fgrant_or_reject\x18\x03 \x01(\x08R\rgrantOrReject\x12@\n\x1dis_selected_based_on_locality\x18\x04 \x01(\x08R\x19isSelectedBasedOnLocality\"\x81\x06\n\x17RequestWorkerLeaseReply\x12\x37\n\x0eworker_address\x18\x01 \x01(\x0b\x32\x10.ray.rpc.AddressR\rworkerAddress\x12G\n\x17retry_at_raylet_address\x18\x02 \x01(\x0b\x32\x10.ray.rpc.AddressR\x14retryAtRayletAddress\x12\x44\n\x10resource_mapping\x18\x03 \x03(\x0b\x32\x19.ray.rpc.ResourceMapEntryR\x0fresourceMapping\x12\x1a\n\x08\x63\x61nceled\x18\x04 \x01(\x08R\x08\x63\x61nceled\x12\x1d\n\nworker_pid\x18\x06 \x01(\rR\tworkerPid\x12\x1a\n\x08rejected\x18\x07 \x01(\x08R\x08rejected\x12=\n\x0eresources_data\x18\x08 \x01(\x0b\x32\x16.ray.rpc.ResourcesDataR\rresourcesData\x12Y\n\x0c\x66\x61ilure_type\x18\t \x01(\x0e\x32\x36.ray.rpc.RequestWorkerLeaseReply.SchedulingFailureTypeR\x0b\x66\x61ilureType\x12<\n\x1ascheduling_failure_message\x18\n \x01(\tR\x18schedulingFailureMessage\"\xee\x01\n\x15SchedulingFailureType\x12\x0e\n\nNOT_FAILED\x10\x00\x12\x15\n\x11SCHEDULING_FAILED\x10\x01\x12\x30\n,SCHEDULING_CANCELLED_PLACEMENT_GROUP_REMOVED\x10\x02\x12\x31\n-SCHEDULING_CANCELLED_RUNTIME_ENV_SETUP_FAILED\x10\x03\x12!\n\x1dSCHEDULING_CANCELLED_INTENDED\x10\x04\x12&\n\"SCHEDULING_CANCELLED_UNSCHEDULABLE\x10\x05\"\x8b\x02\n\x16PrestartWorkersRequest\x12-\n\x08language\x18\x01 \x01(\x0e\x32\x11.ray.rpc.LanguageR\x08language\x12\x1a\n\x06job_id\x18\x02 \x01(\x0cH\x00R\x05jobId\x88\x01\x01\x12\x41\n\x10runtime_env_info\x18\x03 \x01(\x0b\x32\x17.ray.rpc.RuntimeEnvInfoR\x0eruntimeEnvInfo\x12\x37\n\x18keep_alive_duration_secs\x18\x04 \x01(\x04R\x15keepAliveDurationSecs\x12\x1f\n\x0bnum_workers\x18\x05 \x01(\x04R\nnumWorkersB\t\n\x07_job_id\"\x16\n\x14PrestartWorkersReply\"S\n\x1dPrepareBundleResourcesRequest\x12\x32\n\x0c\x62undle_specs\x18\x01 \x03(\x0b\x32\x0f.ray.rpc.BundleR\x0b\x62undleSpecs\"7\n\x1bPrepareBundleResourcesReply\x12\x18\n\x07success\x18\x01 \x01(\x08R\x07success\"R\n\x1c\x43ommitBundleResourcesRequest\x12\x32\n\x0c\x62undle_specs\x18\x01 \x03(\x0b\x32\x0f.ray.rpc.BundleR\x0b\x62undleSpecs\"\x1c\n\x1a\x43ommitBundleResourcesReply\"P\n\x1c\x43\x61ncelResourceReserveRequest\x12\x30\n\x0b\x62undle_spec\x18\x01 \x01(\x0b\x32\x0f.ray.rpc.BundleR\nbundleSpec\"\x1c\n\x1a\x43\x61ncelResourceReserveReply\"\xec\x01\n\x13ReturnWorkerRequest\x12\x1f\n\x0bworker_port\x18\x01 \x01(\x05R\nworkerPort\x12\x1b\n\tworker_id\x18\x02 \x01(\x0cR\x08workerId\x12+\n\x11\x64isconnect_worker\x18\x03 \x01(\x08R\x10\x64isconnectWorker\x12%\n\x0eworker_exiting\x18\x04 \x01(\x08R\rworkerExiting\x12\x43\n\x1e\x64isconnect_worker_error_detail\x18\x05 \x01(\tR\x1b\x64isconnectWorkerErrorDetail\"\x13\n\x11ReturnWorkerReply\"M\n ReleaseUnusedActorWorkersRequest\x12)\n\x11worker_ids_in_use\x18\x01 \x03(\x0cR\x0eworkerIdsInUse\" \n\x1eReleaseUnusedActorWorkersReply\"3\n\x15ShutdownRayletRequest\x12\x1a\n\x08graceful\x18\x01 \x01(\x08R\x08graceful\"\x15\n\x13ShutdownRayletReply\"3\n\x18\x43\x61ncelWorkerLeaseRequest\x12\x17\n\x07task_id\x18\x01 \x01(\x0cR\x06taskId\"2\n\x16\x43\x61ncelWorkerLeaseReply\x12\x18\n\x07success\x18\x01 \x01(\x08R\x07success\"\xa4\x01\n\x13PinObjectIDsRequest\x12\x35\n\rowner_address\x18\x01 \x01(\x0b\x32\x10.ray.rpc.AddressR\x0cownerAddress\x12\x1d\n\nobject_ids\x18\x02 \x03(\x0cR\tobjectIds\x12&\n\x0cgenerator_id\x18\x03 \x01(\x0cH\x00R\x0bgeneratorId\x88\x01\x01\x42\x0f\n\r_generator_id\"1\n\x11PinObjectIDsReply\x12\x1c\n\tsuccesses\x18\x01 \x03(\x08R\tsuccesses\"E\n\x13GetNodeStatsRequest\x12.\n\x13include_memory_info\x18\x01 \x01(\x08R\x11includeMemoryInfo\"\xf2\x06\n\x10ObjectStoreStats\x12+\n\x12spill_time_total_s\x18\x01 \x01(\x01R\x0fspillTimeTotalS\x12.\n\x13spilled_bytes_total\x18\x02 \x01(\x03R\x11spilledBytesTotal\x12\x32\n\x15spilled_objects_total\x18\x03 \x01(\x03R\x13spilledObjectsTotal\x12/\n\x14restore_time_total_s\x18\x04 \x01(\x01R\x11restoreTimeTotalS\x12\x30\n\x14restored_bytes_total\x18\x05 \x01(\x03R\x12restoredBytesTotal\x12\x34\n\x16restored_objects_total\x18\x06 \x01(\x03R\x14restoredObjectsTotal\x12\x35\n\x17object_store_bytes_used\x18\x07 \x01(\x03R\x14objectStoreBytesUsed\x12\x37\n\x18object_store_bytes_avail\x18\x08 \x01(\x03R\x15objectStoreBytesAvail\x12\x44\n\x1fobject_store_bytes_primary_copy\x18\t \x01(\x03R\x1bobjectStoreBytesPrimaryCopy\x12=\n\x1bobject_store_bytes_fallback\x18\n \x01(\x03R\x18objectStoreBytesFallback\x12*\n\x11num_local_objects\x18\x0b \x01(\x03R\x0fnumLocalObjects\x12%\n\x0e\x63onsumed_bytes\x18\x0c \x01(\x03R\rconsumedBytes\x12.\n\x13object_pulls_queued\x18\r \x01(\x08R\x11objectPullsQueued\x12\x44\n\x1fnum_object_store_primary_copies\x18\x0e \x01(\x03R\x1bnumObjectStorePrimaryCopies\x12<\n\x1a\x63umulative_created_objects\x18\x0f \x01(\x03R\x18\x63umulativeCreatedObjects\x12\x38\n\x18\x63umulative_created_bytes\x18\x10 \x01(\x03R\x16\x63umulativeCreatedBytes\"\xb8\x01\n\x11GetNodeStatsReply\x12\x46\n\x12\x63ore_workers_stats\x18\x01 \x03(\x0b\x32\x18.ray.rpc.CoreWorkerStatsR\x10\x63oreWorkersStats\x12\x1f\n\x0bnum_workers\x18\x03 \x01(\rR\nnumWorkers\x12:\n\x0bstore_stats\x18\x06 \x01(\x0b\x32\x19.ray.rpc.ObjectStoreStatsR\nstoreStats\"\x11\n\x0fGlobalGCRequest\"\x0f\n\rGlobalGCReply\"O\n\x1d\x46ormatGlobalMemoryInfoRequest\x12.\n\x13include_memory_info\x18\x01 \x01(\x08R\x11includeMemoryInfo\"\x80\x01\n\x1b\x46ormatGlobalMemoryInfoReply\x12%\n\x0ememory_summary\x18\x01 \x01(\tR\rmemorySummary\x12:\n\x0bstore_stats\x18\x02 \x01(\x0b\x32\x19.ray.rpc.ObjectStoreStatsR\nstoreStats\"T\n\x1bReleaseUnusedBundlesRequest\x12\x35\n\x0e\x62undles_in_use\x18\x01 \x03(\x0b\x32\x0f.ray.rpc.BundleR\x0c\x62undlesInUse\"\x1b\n\x19ReleaseUnusedBundlesReply\"\x18\n\x16GetSystemConfigRequest\";\n\x14GetSystemConfigReply\x12#\n\rsystem_config\x18\x01 \x01(\tR\x0csystemConfig\"<\n\x15GetObjectsInfoRequest\x12\x19\n\x05limit\x18\x01 \x01(\x03H\x00R\x05limit\x88\x01\x01\x42\x08\n\x06_limit\"s\n\x13GetObjectsInfoReply\x12\x46\n\x12\x63ore_workers_stats\x18\x01 \x03(\x0b\x32\x18.ray.rpc.CoreWorkerStatsR\x10\x63oreWorkersStats\x12\x14\n\x05total\x18\x02 \x01(\x03R\x05total\"\x18\n\x16GetResourceLoadRequest\"L\n\x14GetResourceLoadReply\x12\x34\n\tresources\x18\x01 \x01(\x0b\x32\x16.ray.rpc.ResourcesDataR\tresources\"\x19\n\x17NotifyGCSRestartRequest\"\x17\n\x15NotifyGCSRestartReply\"5\n\x1aGetTaskFailureCauseRequest\x12\x17\n\x07task_id\x18\x01 \x01(\x0cR\x06taskId\"\xa1\x01\n\x18GetTaskFailureCauseReply\x12?\n\rfailure_cause\x18\x01 \x01(\x0b\x32\x15.ray.rpc.RayErrorInfoH\x00R\x0c\x66\x61ilureCause\x88\x01\x01\x12\x32\n\x15\x66\x61il_task_immediately\x18\x02 \x01(\x08R\x13\x66\x61ilTaskImmediatelyB\x10\n\x0e_failure_cause\"\xac\x01\n\x12\x44rainRayletRequest\x12;\n\x06reason\x18\x01 \x01(\x0e\x32#.ray.rpc.autoscaler.DrainNodeReasonR\x06reason\x12%\n\x0ereason_message\x18\x02 \x01(\tR\rreasonMessage\x12\x32\n\x15\x64\x65\x61\x64line_timestamp_ms\x18\x03 \x01(\x03R\x13\x64\x65\x61\x64lineTimestampMs\"m\n\x10\x44rainRayletReply\x12\x1f\n\x0bis_accepted\x18\x01 \x01(\x08R\nisAccepted\x12\x38\n\x18rejection_reason_message\x18\x02 \x01(\tR\x16rejectionReasonMessage\"\x93\x01\n\x1cRegisterMutableObjectRequest\x12(\n\x10writer_object_id\x18\x01 \x01(\x0cR\x0ewriterObjectId\x12\x1f\n\x0bnum_readers\x18\x02 \x01(\x03R\nnumReaders\x12(\n\x10reader_object_id\x18\x03 \x01(\x0cR\x0ereaderObjectId\"\x1c\n\x1aRegisterMutableObjectReply\"\xed\x01\n\x18PushMutableObjectRequest\x12(\n\x10writer_object_id\x18\x01 \x01(\x0cR\x0ewriterObjectId\x12&\n\x0ftotal_data_size\x18\x02 \x01(\x04R\rtotalDataSize\x12.\n\x13total_metadata_size\x18\x03 \x01(\x04R\x11totalMetadataSize\x12\x16\n\x06offset\x18\x04 \x01(\x04R\x06offset\x12\x1d\n\nchunk_size\x18\x05 \x01(\x04R\tchunkSize\x12\x18\n\x07payload\x18\x06 \x01(\x0cR\x07payload\",\n\x16PushMutableObjectReply\x12\x12\n\x04\x64one\x18\x01 \x01(\x08R\x04\x64one\"7\n\x18IsLocalWorkerDeadRequest\x12\x1b\n\tworker_id\x18\x01 \x01(\x0cR\x08workerId\"1\n\x16IsLocalWorkerDeadReply\x12\x17\n\x07is_dead\x18\x01 \x01(\x08R\x06isDead2\xdd\x10\n\x12NodeManagerService\x12T\n\x10NotifyGCSRestart\x12 .ray.rpc.NotifyGCSRestartRequest\x1a\x1e.ray.rpc.NotifyGCSRestartReply\x12Q\n\x0fGetResourceLoad\x12\x1f.ray.rpc.GetResourceLoadRequest\x1a\x1d.ray.rpc.GetResourceLoadReply\x12Z\n\x12RequestWorkerLease\x12\".ray.rpc.RequestWorkerLeaseRequest\x1a .ray.rpc.RequestWorkerLeaseReply\x12Q\n\x0fPrestartWorkers\x12\x1f.ray.rpc.PrestartWorkersRequest\x1a\x1d.ray.rpc.PrestartWorkersReply\x12]\n\x13ReportWorkerBacklog\x12#.ray.rpc.ReportWorkerBacklogRequest\x1a!.ray.rpc.ReportWorkerBacklogReply\x12H\n\x0cReturnWorker\x12\x1c.ray.rpc.ReturnWorkerRequest\x1a\x1a.ray.rpc.ReturnWorkerReply\x12o\n\x19ReleaseUnusedActorWorkers\x12).ray.rpc.ReleaseUnusedActorWorkersRequest\x1a\'.ray.rpc.ReleaseUnusedActorWorkersReply\x12N\n\x0eShutdownRaylet\x12\x1e.ray.rpc.ShutdownRayletRequest\x1a\x1c.ray.rpc.ShutdownRayletReply\x12\x45\n\x0b\x44rainRaylet\x12\x1b.ray.rpc.DrainRayletRequest\x1a\x19.ray.rpc.DrainRayletReply\x12\x66\n\x16PrepareBundleResources\x12&.ray.rpc.PrepareBundleResourcesRequest\x1a$.ray.rpc.PrepareBundleResourcesReply\x12\x63\n\x15\x43ommitBundleResources\x12%.ray.rpc.CommitBundleResourcesRequest\x1a#.ray.rpc.CommitBundleResourcesReply\x12\x63\n\x15\x43\x61ncelResourceReserve\x12%.ray.rpc.CancelResourceReserveRequest\x1a#.ray.rpc.CancelResourceReserveReply\x12W\n\x11\x43\x61ncelWorkerLease\x12!.ray.rpc.CancelWorkerLeaseRequest\x1a\x1f.ray.rpc.CancelWorkerLeaseReply\x12H\n\x0cPinObjectIDs\x12\x1c.ray.rpc.PinObjectIDsRequest\x1a\x1a.ray.rpc.PinObjectIDsReply\x12H\n\x0cGetNodeStats\x12\x1c.ray.rpc.GetNodeStatsRequest\x1a\x1a.ray.rpc.GetNodeStatsReply\x12<\n\x08GlobalGC\x12\x18.ray.rpc.GlobalGCRequest\x1a\x16.ray.rpc.GlobalGCReply\x12\x66\n\x16\x46ormatGlobalMemoryInfo\x12&.ray.rpc.FormatGlobalMemoryInfoRequest\x1a$.ray.rpc.FormatGlobalMemoryInfoReply\x12`\n\x14ReleaseUnusedBundles\x12$.ray.rpc.ReleaseUnusedBundlesRequest\x1a\".ray.rpc.ReleaseUnusedBundlesReply\x12Q\n\x0fGetSystemConfig\x12\x1f.ray.rpc.GetSystemConfigRequest\x1a\x1d.ray.rpc.GetSystemConfigReply\x12N\n\x0eGetObjectsInfo\x12\x1e.ray.rpc.GetObjectsInfoRequest\x1a\x1c.ray.rpc.GetObjectsInfoReply\x12]\n\x13GetTaskFailureCause\x12#.ray.rpc.GetTaskFailureCauseRequest\x1a!.ray.rpc.GetTaskFailureCauseReply\x12\x63\n\x15RegisterMutableObject\x12%.ray.rpc.RegisterMutableObjectRequest\x1a#.ray.rpc.RegisterMutableObjectReply\x12W\n\x11PushMutableObject\x12!.ray.rpc.PushMutableObjectRequest\x1a\x1f.ray.rpc.PushMutableObjectReply\x12W\n\x11IsLocalWorkerDead\x12!.ray.rpc.IsLocalWorkerDeadRequest\x1a\x1f.ray.rpc.IsLocalWorkerDeadReplyb\x06proto3')
+
+
+
+_WORKERBACKLOGREPORT = DESCRIPTOR.message_types_by_name['WorkerBacklogReport']
+_REPORTWORKERBACKLOGREQUEST = DESCRIPTOR.message_types_by_name['ReportWorkerBacklogRequest']
+_REPORTWORKERBACKLOGREPLY = DESCRIPTOR.message_types_by_name['ReportWorkerBacklogReply']
+_REQUESTWORKERLEASEREQUEST = DESCRIPTOR.message_types_by_name['RequestWorkerLeaseRequest']
+_REQUESTWORKERLEASEREPLY = DESCRIPTOR.message_types_by_name['RequestWorkerLeaseReply']
+_PRESTARTWORKERSREQUEST = DESCRIPTOR.message_types_by_name['PrestartWorkersRequest']
+_PRESTARTWORKERSREPLY = DESCRIPTOR.message_types_by_name['PrestartWorkersReply']
+_PREPAREBUNDLERESOURCESREQUEST = DESCRIPTOR.message_types_by_name['PrepareBundleResourcesRequest']
+_PREPAREBUNDLERESOURCESREPLY = DESCRIPTOR.message_types_by_name['PrepareBundleResourcesReply']
+_COMMITBUNDLERESOURCESREQUEST = DESCRIPTOR.message_types_by_name['CommitBundleResourcesRequest']
+_COMMITBUNDLERESOURCESREPLY = DESCRIPTOR.message_types_by_name['CommitBundleResourcesReply']
+_CANCELRESOURCERESERVEREQUEST = DESCRIPTOR.message_types_by_name['CancelResourceReserveRequest']
+_CANCELRESOURCERESERVEREPLY = DESCRIPTOR.message_types_by_name['CancelResourceReserveReply']
+_RETURNWORKERREQUEST = DESCRIPTOR.message_types_by_name['ReturnWorkerRequest']
+_RETURNWORKERREPLY = DESCRIPTOR.message_types_by_name['ReturnWorkerReply']
+_RELEASEUNUSEDACTORWORKERSREQUEST = DESCRIPTOR.message_types_by_name['ReleaseUnusedActorWorkersRequest']
+_RELEASEUNUSEDACTORWORKERSREPLY = DESCRIPTOR.message_types_by_name['ReleaseUnusedActorWorkersReply']
+_SHUTDOWNRAYLETREQUEST = DESCRIPTOR.message_types_by_name['ShutdownRayletRequest']
+_SHUTDOWNRAYLETREPLY = DESCRIPTOR.message_types_by_name['ShutdownRayletReply']
+_CANCELWORKERLEASEREQUEST = DESCRIPTOR.message_types_by_name['CancelWorkerLeaseRequest']
+_CANCELWORKERLEASEREPLY = DESCRIPTOR.message_types_by_name['CancelWorkerLeaseReply']
+_PINOBJECTIDSREQUEST = DESCRIPTOR.message_types_by_name['PinObjectIDsRequest']
+_PINOBJECTIDSREPLY = DESCRIPTOR.message_types_by_name['PinObjectIDsReply']
+_GETNODESTATSREQUEST = DESCRIPTOR.message_types_by_name['GetNodeStatsRequest']
+_OBJECTSTORESTATS = DESCRIPTOR.message_types_by_name['ObjectStoreStats']
+_GETNODESTATSREPLY = DESCRIPTOR.message_types_by_name['GetNodeStatsReply']
+_GLOBALGCREQUEST = DESCRIPTOR.message_types_by_name['GlobalGCRequest']
+_GLOBALGCREPLY = DESCRIPTOR.message_types_by_name['GlobalGCReply']
+_FORMATGLOBALMEMORYINFOREQUEST = DESCRIPTOR.message_types_by_name['FormatGlobalMemoryInfoRequest']
+_FORMATGLOBALMEMORYINFOREPLY = DESCRIPTOR.message_types_by_name['FormatGlobalMemoryInfoReply']
+_RELEASEUNUSEDBUNDLESREQUEST = DESCRIPTOR.message_types_by_name['ReleaseUnusedBundlesRequest']
+_RELEASEUNUSEDBUNDLESREPLY = DESCRIPTOR.message_types_by_name['ReleaseUnusedBundlesReply']
+_GETSYSTEMCONFIGREQUEST = DESCRIPTOR.message_types_by_name['GetSystemConfigRequest']
+_GETSYSTEMCONFIGREPLY = DESCRIPTOR.message_types_by_name['GetSystemConfigReply']
+_GETOBJECTSINFOREQUEST = DESCRIPTOR.message_types_by_name['GetObjectsInfoRequest']
+_GETOBJECTSINFOREPLY = DESCRIPTOR.message_types_by_name['GetObjectsInfoReply']
+_GETRESOURCELOADREQUEST = DESCRIPTOR.message_types_by_name['GetResourceLoadRequest']
+_GETRESOURCELOADREPLY = DESCRIPTOR.message_types_by_name['GetResourceLoadReply']
+_NOTIFYGCSRESTARTREQUEST = DESCRIPTOR.message_types_by_name['NotifyGCSRestartRequest']
+_NOTIFYGCSRESTARTREPLY = DESCRIPTOR.message_types_by_name['NotifyGCSRestartReply']
+_GETTASKFAILURECAUSEREQUEST = DESCRIPTOR.message_types_by_name['GetTaskFailureCauseRequest']
+_GETTASKFAILURECAUSEREPLY = DESCRIPTOR.message_types_by_name['GetTaskFailureCauseReply']
+_DRAINRAYLETREQUEST = DESCRIPTOR.message_types_by_name['DrainRayletRequest']
+_DRAINRAYLETREPLY = DESCRIPTOR.message_types_by_name['DrainRayletReply']
+_REGISTERMUTABLEOBJECTREQUEST = DESCRIPTOR.message_types_by_name['RegisterMutableObjectRequest']
+_REGISTERMUTABLEOBJECTREPLY = DESCRIPTOR.message_types_by_name['RegisterMutableObjectReply']
+_PUSHMUTABLEOBJECTREQUEST = DESCRIPTOR.message_types_by_name['PushMutableObjectRequest']
+_PUSHMUTABLEOBJECTREPLY = DESCRIPTOR.message_types_by_name['PushMutableObjectReply']
+_ISLOCALWORKERDEADREQUEST = DESCRIPTOR.message_types_by_name['IsLocalWorkerDeadRequest']
+_ISLOCALWORKERDEADREPLY = DESCRIPTOR.message_types_by_name['IsLocalWorkerDeadReply']
+_REQUESTWORKERLEASEREPLY_SCHEDULINGFAILURETYPE = _REQUESTWORKERLEASEREPLY.enum_types_by_name['SchedulingFailureType']
+WorkerBacklogReport = _reflection.GeneratedProtocolMessageType('WorkerBacklogReport', (_message.Message,), {
+  'DESCRIPTOR' : _WORKERBACKLOGREPORT,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.WorkerBacklogReport)
+  })
+_sym_db.RegisterMessage(WorkerBacklogReport)
+
+ReportWorkerBacklogRequest = _reflection.GeneratedProtocolMessageType('ReportWorkerBacklogRequest', (_message.Message,), {
+  'DESCRIPTOR' : _REPORTWORKERBACKLOGREQUEST,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.ReportWorkerBacklogRequest)
+  })
+_sym_db.RegisterMessage(ReportWorkerBacklogRequest)
+
+ReportWorkerBacklogReply = _reflection.GeneratedProtocolMessageType('ReportWorkerBacklogReply', (_message.Message,), {
+  'DESCRIPTOR' : _REPORTWORKERBACKLOGREPLY,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.ReportWorkerBacklogReply)
+  })
+_sym_db.RegisterMessage(ReportWorkerBacklogReply)
+
+RequestWorkerLeaseRequest = _reflection.GeneratedProtocolMessageType('RequestWorkerLeaseRequest', (_message.Message,), {
+  'DESCRIPTOR' : _REQUESTWORKERLEASEREQUEST,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.RequestWorkerLeaseRequest)
+  })
+_sym_db.RegisterMessage(RequestWorkerLeaseRequest)
+
+RequestWorkerLeaseReply = _reflection.GeneratedProtocolMessageType('RequestWorkerLeaseReply', (_message.Message,), {
+  'DESCRIPTOR' : _REQUESTWORKERLEASEREPLY,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.RequestWorkerLeaseReply)
+  })
+_sym_db.RegisterMessage(RequestWorkerLeaseReply)
+
+PrestartWorkersRequest = _reflection.GeneratedProtocolMessageType('PrestartWorkersRequest', (_message.Message,), {
+  'DESCRIPTOR' : _PRESTARTWORKERSREQUEST,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.PrestartWorkersRequest)
+  })
+_sym_db.RegisterMessage(PrestartWorkersRequest)
+
+PrestartWorkersReply = _reflection.GeneratedProtocolMessageType('PrestartWorkersReply', (_message.Message,), {
+  'DESCRIPTOR' : _PRESTARTWORKERSREPLY,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.PrestartWorkersReply)
+  })
+_sym_db.RegisterMessage(PrestartWorkersReply)
+
+PrepareBundleResourcesRequest = _reflection.GeneratedProtocolMessageType('PrepareBundleResourcesRequest', (_message.Message,), {
+  'DESCRIPTOR' : _PREPAREBUNDLERESOURCESREQUEST,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.PrepareBundleResourcesRequest)
+  })
+_sym_db.RegisterMessage(PrepareBundleResourcesRequest)
+
+PrepareBundleResourcesReply = _reflection.GeneratedProtocolMessageType('PrepareBundleResourcesReply', (_message.Message,), {
+  'DESCRIPTOR' : _PREPAREBUNDLERESOURCESREPLY,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.PrepareBundleResourcesReply)
+  })
+_sym_db.RegisterMessage(PrepareBundleResourcesReply)
+
+CommitBundleResourcesRequest = _reflection.GeneratedProtocolMessageType('CommitBundleResourcesRequest', (_message.Message,), {
+  'DESCRIPTOR' : _COMMITBUNDLERESOURCESREQUEST,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.CommitBundleResourcesRequest)
+  })
+_sym_db.RegisterMessage(CommitBundleResourcesRequest)
+
+CommitBundleResourcesReply = _reflection.GeneratedProtocolMessageType('CommitBundleResourcesReply', (_message.Message,), {
+  'DESCRIPTOR' : _COMMITBUNDLERESOURCESREPLY,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.CommitBundleResourcesReply)
+  })
+_sym_db.RegisterMessage(CommitBundleResourcesReply)
+
+CancelResourceReserveRequest = _reflection.GeneratedProtocolMessageType('CancelResourceReserveRequest', (_message.Message,), {
+  'DESCRIPTOR' : _CANCELRESOURCERESERVEREQUEST,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.CancelResourceReserveRequest)
+  })
+_sym_db.RegisterMessage(CancelResourceReserveRequest)
+
+CancelResourceReserveReply = _reflection.GeneratedProtocolMessageType('CancelResourceReserveReply', (_message.Message,), {
+  'DESCRIPTOR' : _CANCELRESOURCERESERVEREPLY,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.CancelResourceReserveReply)
+  })
+_sym_db.RegisterMessage(CancelResourceReserveReply)
+
+ReturnWorkerRequest = _reflection.GeneratedProtocolMessageType('ReturnWorkerRequest', (_message.Message,), {
+  'DESCRIPTOR' : _RETURNWORKERREQUEST,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.ReturnWorkerRequest)
+  })
+_sym_db.RegisterMessage(ReturnWorkerRequest)
+
+ReturnWorkerReply = _reflection.GeneratedProtocolMessageType('ReturnWorkerReply', (_message.Message,), {
+  'DESCRIPTOR' : _RETURNWORKERREPLY,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.ReturnWorkerReply)
+  })
+_sym_db.RegisterMessage(ReturnWorkerReply)
+
+ReleaseUnusedActorWorkersRequest = _reflection.GeneratedProtocolMessageType('ReleaseUnusedActorWorkersRequest', (_message.Message,), {
+  'DESCRIPTOR' : _RELEASEUNUSEDACTORWORKERSREQUEST,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.ReleaseUnusedActorWorkersRequest)
+  })
+_sym_db.RegisterMessage(ReleaseUnusedActorWorkersRequest)
+
+ReleaseUnusedActorWorkersReply = _reflection.GeneratedProtocolMessageType('ReleaseUnusedActorWorkersReply', (_message.Message,), {
+  'DESCRIPTOR' : _RELEASEUNUSEDACTORWORKERSREPLY,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.ReleaseUnusedActorWorkersReply)
+  })
+_sym_db.RegisterMessage(ReleaseUnusedActorWorkersReply)
+
+ShutdownRayletRequest = _reflection.GeneratedProtocolMessageType('ShutdownRayletRequest', (_message.Message,), {
+  'DESCRIPTOR' : _SHUTDOWNRAYLETREQUEST,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.ShutdownRayletRequest)
+  })
+_sym_db.RegisterMessage(ShutdownRayletRequest)
+
+ShutdownRayletReply = _reflection.GeneratedProtocolMessageType('ShutdownRayletReply', (_message.Message,), {
+  'DESCRIPTOR' : _SHUTDOWNRAYLETREPLY,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.ShutdownRayletReply)
+  })
+_sym_db.RegisterMessage(ShutdownRayletReply)
+
+CancelWorkerLeaseRequest = _reflection.GeneratedProtocolMessageType('CancelWorkerLeaseRequest', (_message.Message,), {
+  'DESCRIPTOR' : _CANCELWORKERLEASEREQUEST,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.CancelWorkerLeaseRequest)
+  })
+_sym_db.RegisterMessage(CancelWorkerLeaseRequest)
+
+CancelWorkerLeaseReply = _reflection.GeneratedProtocolMessageType('CancelWorkerLeaseReply', (_message.Message,), {
+  'DESCRIPTOR' : _CANCELWORKERLEASEREPLY,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.CancelWorkerLeaseReply)
+  })
+_sym_db.RegisterMessage(CancelWorkerLeaseReply)
+
+PinObjectIDsRequest = _reflection.GeneratedProtocolMessageType('PinObjectIDsRequest', (_message.Message,), {
+  'DESCRIPTOR' : _PINOBJECTIDSREQUEST,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.PinObjectIDsRequest)
+  })
+_sym_db.RegisterMessage(PinObjectIDsRequest)
+
+PinObjectIDsReply = _reflection.GeneratedProtocolMessageType('PinObjectIDsReply', (_message.Message,), {
+  'DESCRIPTOR' : _PINOBJECTIDSREPLY,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.PinObjectIDsReply)
+  })
+_sym_db.RegisterMessage(PinObjectIDsReply)
+
+GetNodeStatsRequest = _reflection.GeneratedProtocolMessageType('GetNodeStatsRequest', (_message.Message,), {
+  'DESCRIPTOR' : _GETNODESTATSREQUEST,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.GetNodeStatsRequest)
+  })
+_sym_db.RegisterMessage(GetNodeStatsRequest)
+
+ObjectStoreStats = _reflection.GeneratedProtocolMessageType('ObjectStoreStats', (_message.Message,), {
+  'DESCRIPTOR' : _OBJECTSTORESTATS,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.ObjectStoreStats)
+  })
+_sym_db.RegisterMessage(ObjectStoreStats)
+
+GetNodeStatsReply = _reflection.GeneratedProtocolMessageType('GetNodeStatsReply', (_message.Message,), {
+  'DESCRIPTOR' : _GETNODESTATSREPLY,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.GetNodeStatsReply)
+  })
+_sym_db.RegisterMessage(GetNodeStatsReply)
+
+GlobalGCRequest = _reflection.GeneratedProtocolMessageType('GlobalGCRequest', (_message.Message,), {
+  'DESCRIPTOR' : _GLOBALGCREQUEST,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.GlobalGCRequest)
+  })
+_sym_db.RegisterMessage(GlobalGCRequest)
+
+GlobalGCReply = _reflection.GeneratedProtocolMessageType('GlobalGCReply', (_message.Message,), {
+  'DESCRIPTOR' : _GLOBALGCREPLY,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.GlobalGCReply)
+  })
+_sym_db.RegisterMessage(GlobalGCReply)
+
+FormatGlobalMemoryInfoRequest = _reflection.GeneratedProtocolMessageType('FormatGlobalMemoryInfoRequest', (_message.Message,), {
+  'DESCRIPTOR' : _FORMATGLOBALMEMORYINFOREQUEST,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.FormatGlobalMemoryInfoRequest)
+  })
+_sym_db.RegisterMessage(FormatGlobalMemoryInfoRequest)
+
+FormatGlobalMemoryInfoReply = _reflection.GeneratedProtocolMessageType('FormatGlobalMemoryInfoReply', (_message.Message,), {
+  'DESCRIPTOR' : _FORMATGLOBALMEMORYINFOREPLY,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.FormatGlobalMemoryInfoReply)
+  })
+_sym_db.RegisterMessage(FormatGlobalMemoryInfoReply)
+
+ReleaseUnusedBundlesRequest = _reflection.GeneratedProtocolMessageType('ReleaseUnusedBundlesRequest', (_message.Message,), {
+  'DESCRIPTOR' : _RELEASEUNUSEDBUNDLESREQUEST,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.ReleaseUnusedBundlesRequest)
+  })
+_sym_db.RegisterMessage(ReleaseUnusedBundlesRequest)
+
+ReleaseUnusedBundlesReply = _reflection.GeneratedProtocolMessageType('ReleaseUnusedBundlesReply', (_message.Message,), {
+  'DESCRIPTOR' : _RELEASEUNUSEDBUNDLESREPLY,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.ReleaseUnusedBundlesReply)
+  })
+_sym_db.RegisterMessage(ReleaseUnusedBundlesReply)
+
+GetSystemConfigRequest = _reflection.GeneratedProtocolMessageType('GetSystemConfigRequest', (_message.Message,), {
+  'DESCRIPTOR' : _GETSYSTEMCONFIGREQUEST,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.GetSystemConfigRequest)
+  })
+_sym_db.RegisterMessage(GetSystemConfigRequest)
+
+GetSystemConfigReply = _reflection.GeneratedProtocolMessageType('GetSystemConfigReply', (_message.Message,), {
+  'DESCRIPTOR' : _GETSYSTEMCONFIGREPLY,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.GetSystemConfigReply)
+  })
+_sym_db.RegisterMessage(GetSystemConfigReply)
+
+GetObjectsInfoRequest = _reflection.GeneratedProtocolMessageType('GetObjectsInfoRequest', (_message.Message,), {
+  'DESCRIPTOR' : _GETOBJECTSINFOREQUEST,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.GetObjectsInfoRequest)
+  })
+_sym_db.RegisterMessage(GetObjectsInfoRequest)
+
+GetObjectsInfoReply = _reflection.GeneratedProtocolMessageType('GetObjectsInfoReply', (_message.Message,), {
+  'DESCRIPTOR' : _GETOBJECTSINFOREPLY,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.GetObjectsInfoReply)
+  })
+_sym_db.RegisterMessage(GetObjectsInfoReply)
+
+GetResourceLoadRequest = _reflection.GeneratedProtocolMessageType('GetResourceLoadRequest', (_message.Message,), {
+  'DESCRIPTOR' : _GETRESOURCELOADREQUEST,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.GetResourceLoadRequest)
+  })
+_sym_db.RegisterMessage(GetResourceLoadRequest)
+
+GetResourceLoadReply = _reflection.GeneratedProtocolMessageType('GetResourceLoadReply', (_message.Message,), {
+  'DESCRIPTOR' : _GETRESOURCELOADREPLY,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.GetResourceLoadReply)
+  })
+_sym_db.RegisterMessage(GetResourceLoadReply)
+
+NotifyGCSRestartRequest = _reflection.GeneratedProtocolMessageType('NotifyGCSRestartRequest', (_message.Message,), {
+  'DESCRIPTOR' : _NOTIFYGCSRESTARTREQUEST,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.NotifyGCSRestartRequest)
+  })
+_sym_db.RegisterMessage(NotifyGCSRestartRequest)
+
+NotifyGCSRestartReply = _reflection.GeneratedProtocolMessageType('NotifyGCSRestartReply', (_message.Message,), {
+  'DESCRIPTOR' : _NOTIFYGCSRESTARTREPLY,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.NotifyGCSRestartReply)
+  })
+_sym_db.RegisterMessage(NotifyGCSRestartReply)
+
+GetTaskFailureCauseRequest = _reflection.GeneratedProtocolMessageType('GetTaskFailureCauseRequest', (_message.Message,), {
+  'DESCRIPTOR' : _GETTASKFAILURECAUSEREQUEST,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.GetTaskFailureCauseRequest)
+  })
+_sym_db.RegisterMessage(GetTaskFailureCauseRequest)
+
+GetTaskFailureCauseReply = _reflection.GeneratedProtocolMessageType('GetTaskFailureCauseReply', (_message.Message,), {
+  'DESCRIPTOR' : _GETTASKFAILURECAUSEREPLY,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.GetTaskFailureCauseReply)
+  })
+_sym_db.RegisterMessage(GetTaskFailureCauseReply)
+
+DrainRayletRequest = _reflection.GeneratedProtocolMessageType('DrainRayletRequest', (_message.Message,), {
+  'DESCRIPTOR' : _DRAINRAYLETREQUEST,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.DrainRayletRequest)
+  })
+_sym_db.RegisterMessage(DrainRayletRequest)
+
+DrainRayletReply = _reflection.GeneratedProtocolMessageType('DrainRayletReply', (_message.Message,), {
+  'DESCRIPTOR' : _DRAINRAYLETREPLY,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.DrainRayletReply)
+  })
+_sym_db.RegisterMessage(DrainRayletReply)
+
+RegisterMutableObjectRequest = _reflection.GeneratedProtocolMessageType('RegisterMutableObjectRequest', (_message.Message,), {
+  'DESCRIPTOR' : _REGISTERMUTABLEOBJECTREQUEST,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.RegisterMutableObjectRequest)
+  })
+_sym_db.RegisterMessage(RegisterMutableObjectRequest)
+
+RegisterMutableObjectReply = _reflection.GeneratedProtocolMessageType('RegisterMutableObjectReply', (_message.Message,), {
+  'DESCRIPTOR' : _REGISTERMUTABLEOBJECTREPLY,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.RegisterMutableObjectReply)
+  })
+_sym_db.RegisterMessage(RegisterMutableObjectReply)
+
+PushMutableObjectRequest = _reflection.GeneratedProtocolMessageType('PushMutableObjectRequest', (_message.Message,), {
+  'DESCRIPTOR' : _PUSHMUTABLEOBJECTREQUEST,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.PushMutableObjectRequest)
+  })
+_sym_db.RegisterMessage(PushMutableObjectRequest)
+
+PushMutableObjectReply = _reflection.GeneratedProtocolMessageType('PushMutableObjectReply', (_message.Message,), {
+  'DESCRIPTOR' : _PUSHMUTABLEOBJECTREPLY,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.PushMutableObjectReply)
+  })
+_sym_db.RegisterMessage(PushMutableObjectReply)
+
+IsLocalWorkerDeadRequest = _reflection.GeneratedProtocolMessageType('IsLocalWorkerDeadRequest', (_message.Message,), {
+  'DESCRIPTOR' : _ISLOCALWORKERDEADREQUEST,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.IsLocalWorkerDeadRequest)
+  })
+_sym_db.RegisterMessage(IsLocalWorkerDeadRequest)
+
+IsLocalWorkerDeadReply = _reflection.GeneratedProtocolMessageType('IsLocalWorkerDeadReply', (_message.Message,), {
+  'DESCRIPTOR' : _ISLOCALWORKERDEADREPLY,
+  '__module__' : 'src.ray.protobuf.node_manager_pb2'
+  # @@protoc_insertion_point(class_scope:ray.rpc.IsLocalWorkerDeadReply)
+  })
+_sym_db.RegisterMessage(IsLocalWorkerDeadReply)
+
+_NODEMANAGERSERVICE = DESCRIPTOR.services_by_name['NodeManagerService']
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  _WORKERBACKLOGREPORT._serialized_start=185
+  _WORKERBACKLOGREPORT._serialized_end=297
+  _REPORTWORKERBACKLOGREQUEST._serialized_start=300
+  _REPORTWORKERBACKLOGREQUEST._serialized_end=428
+  _REPORTWORKERBACKLOGREPLY._serialized_start=430
+  _REPORTWORKERBACKLOGREPLY._serialized_end=456
+  _REQUESTWORKERLEASEREQUEST._serialized_start=459
+  _REQUESTWORKERLEASEREQUEST._serialized_end=683
+  _REQUESTWORKERLEASEREPLY._serialized_start=686
+  _REQUESTWORKERLEASEREPLY._serialized_end=1455
+  _REQUESTWORKERLEASEREPLY_SCHEDULINGFAILURETYPE._serialized_start=1217
+  _REQUESTWORKERLEASEREPLY_SCHEDULINGFAILURETYPE._serialized_end=1455
+  _PRESTARTWORKERSREQUEST._serialized_start=1458
+  _PRESTARTWORKERSREQUEST._serialized_end=1725
+  _PRESTARTWORKERSREPLY._serialized_start=1727
+  _PRESTARTWORKERSREPLY._serialized_end=1749
+  _PREPAREBUNDLERESOURCESREQUEST._serialized_start=1751
+  _PREPAREBUNDLERESOURCESREQUEST._serialized_end=1834
+  _PREPAREBUNDLERESOURCESREPLY._serialized_start=1836
+  _PREPAREBUNDLERESOURCESREPLY._serialized_end=1891
+  _COMMITBUNDLERESOURCESREQUEST._serialized_start=1893
+  _COMMITBUNDLERESOURCESREQUEST._serialized_end=1975
+  _COMMITBUNDLERESOURCESREPLY._serialized_start=1977
+  _COMMITBUNDLERESOURCESREPLY._serialized_end=2005
+  _CANCELRESOURCERESERVEREQUEST._serialized_start=2007
+  _CANCELRESOURCERESERVEREQUEST._serialized_end=2087
+  _CANCELRESOURCERESERVEREPLY._serialized_start=2089
+  _CANCELRESOURCERESERVEREPLY._serialized_end=2117
+  _RETURNWORKERREQUEST._serialized_start=2120
+  _RETURNWORKERREQUEST._serialized_end=2356
+  _RETURNWORKERREPLY._serialized_start=2358
+  _RETURNWORKERREPLY._serialized_end=2377
+  _RELEASEUNUSEDACTORWORKERSREQUEST._serialized_start=2379
+  _RELEASEUNUSEDACTORWORKERSREQUEST._serialized_end=2456
+  _RELEASEUNUSEDACTORWORKERSREPLY._serialized_start=2458
+  _RELEASEUNUSEDACTORWORKERSREPLY._serialized_end=2490
+  _SHUTDOWNRAYLETREQUEST._serialized_start=2492
+  _SHUTDOWNRAYLETREQUEST._serialized_end=2543
+  _SHUTDOWNRAYLETREPLY._serialized_start=2545
+  _SHUTDOWNRAYLETREPLY._serialized_end=2566
+  _CANCELWORKERLEASEREQUEST._serialized_start=2568
+  _CANCELWORKERLEASEREQUEST._serialized_end=2619
+  _CANCELWORKERLEASEREPLY._serialized_start=2621
+  _CANCELWORKERLEASEREPLY._serialized_end=2671
+  _PINOBJECTIDSREQUEST._serialized_start=2674
+  _PINOBJECTIDSREQUEST._serialized_end=2838
+  _PINOBJECTIDSREPLY._serialized_start=2840
+  _PINOBJECTIDSREPLY._serialized_end=2889
+  _GETNODESTATSREQUEST._serialized_start=2891
+  _GETNODESTATSREQUEST._serialized_end=2960
+  _OBJECTSTORESTATS._serialized_start=2963
+  _OBJECTSTORESTATS._serialized_end=3845
+  _GETNODESTATSREPLY._serialized_start=3848
+  _GETNODESTATSREPLY._serialized_end=4032
+  _GLOBALGCREQUEST._serialized_start=4034
+  _GLOBALGCREQUEST._serialized_end=4051
+  _GLOBALGCREPLY._serialized_start=4053
+  _GLOBALGCREPLY._serialized_end=4068
+  _FORMATGLOBALMEMORYINFOREQUEST._serialized_start=4070
+  _FORMATGLOBALMEMORYINFOREQUEST._serialized_end=4149
+  _FORMATGLOBALMEMORYINFOREPLY._serialized_start=4152
+  _FORMATGLOBALMEMORYINFOREPLY._serialized_end=4280
+  _RELEASEUNUSEDBUNDLESREQUEST._serialized_start=4282
+  _RELEASEUNUSEDBUNDLESREQUEST._serialized_end=4366
+  _RELEASEUNUSEDBUNDLESREPLY._serialized_start=4368
+  _RELEASEUNUSEDBUNDLESREPLY._serialized_end=4395
+  _GETSYSTEMCONFIGREQUEST._serialized_start=4397
+  _GETSYSTEMCONFIGREQUEST._serialized_end=4421
+  _GETSYSTEMCONFIGREPLY._serialized_start=4423
+  _GETSYSTEMCONFIGREPLY._serialized_end=4482
+  _GETOBJECTSINFOREQUEST._serialized_start=4484
+  _GETOBJECTSINFOREQUEST._serialized_end=4544
+  _GETOBJECTSINFOREPLY._serialized_start=4546
+  _GETOBJECTSINFOREPLY._serialized_end=4661
+  _GETRESOURCELOADREQUEST._serialized_start=4663
+  _GETRESOURCELOADREQUEST._serialized_end=4687
+  _GETRESOURCELOADREPLY._serialized_start=4689
+  _GETRESOURCELOADREPLY._serialized_end=4765
+  _NOTIFYGCSRESTARTREQUEST._serialized_start=4767
+  _NOTIFYGCSRESTARTREQUEST._serialized_end=4792
+  _NOTIFYGCSRESTARTREPLY._serialized_start=4794
+  _NOTIFYGCSRESTARTREPLY._serialized_end=4817
+  _GETTASKFAILURECAUSEREQUEST._serialized_start=4819
+  _GETTASKFAILURECAUSEREQUEST._serialized_end=4872
+  _GETTASKFAILURECAUSEREPLY._serialized_start=4875
+  _GETTASKFAILURECAUSEREPLY._serialized_end=5036
+  _DRAINRAYLETREQUEST._serialized_start=5039
+  _DRAINRAYLETREQUEST._serialized_end=5211
+  _DRAINRAYLETREPLY._serialized_start=5213
+  _DRAINRAYLETREPLY._serialized_end=5322
+  _REGISTERMUTABLEOBJECTREQUEST._serialized_start=5325
+  _REGISTERMUTABLEOBJECTREQUEST._serialized_end=5472
+  _REGISTERMUTABLEOBJECTREPLY._serialized_start=5474
+  _REGISTERMUTABLEOBJECTREPLY._serialized_end=5502
+  _PUSHMUTABLEOBJECTREQUEST._serialized_start=5505
+  _PUSHMUTABLEOBJECTREQUEST._serialized_end=5742
+  _PUSHMUTABLEOBJECTREPLY._serialized_start=5744
+  _PUSHMUTABLEOBJECTREPLY._serialized_end=5788
+  _ISLOCALWORKERDEADREQUEST._serialized_start=5790
+  _ISLOCALWORKERDEADREQUEST._serialized_end=5845
+  _ISLOCALWORKERDEADREPLY._serialized_start=5847
+  _ISLOCALWORKERDEADREPLY._serialized_end=5896
+  _NODEMANAGERSERVICE._serialized_start=5899
+  _NODEMANAGERSERVICE._serialized_end=8040
+# @@protoc_insertion_point(module_scope)

.venv/lib/python3.11/site-packages/ray/core/generated/resource_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

.venv/lib/python3.11/site-packages/ray/dashboard/client/build/speedscope-1.5.3/README

@@ -0,0 +1,2 @@
+This is a self-contained release of https://github.com/jlfwong/speedscope.
+To use it, open index.html in Chrome or Firefox.

.venv/lib/python3.11/site-packages/ray/dashboard/client/build/speedscope-1.5.3/file-format-schema.json

@@ -0,0 +1,324 @@
+{
+    "$schema": "http://json-schema.org/draft-07/schema#",
+    "definitions": {
+        "FileFormat.Profile": {
+            "anyOf": [
+                {
+                    "$ref": "#/definitions/FileFormat.EventedProfile"
+                },
+                {
+                    "$ref": "#/definitions/FileFormat.SampledProfile"
+                }
+            ]
+        },
+        "FileFormat.File": {
+            "title": "FileFormat.File",
+            "type": "object",
+            "properties": {
+                "$schema": {
+                    "type": "string",
+                    "enum": [
+                        "https://www.speedscope.app/file-format-schema.json"
+                    ],
+                    "title": "$schema"
+                },
+                "shared": {
+                    "type": "object",
+                    "properties": {
+                        "frames": {
+                            "type": "array",
+                            "items": {
+                                "$ref": "#/definitions/FileFormat.Frame"
+                            },
+                            "title": "frames"
+                        }
+                    },
+                    "required": [
+                        "frames"
+                    ],
+                    "title": "shared"
+                },
+                "profiles": {
+                    "type": "array",
+                    "items": {
+                        "anyOf": [
+                            {
+                                "$ref": "#/definitions/FileFormat.EventedProfile"
+                            },
+                            {
+                                "$ref": "#/definitions/FileFormat.SampledProfile"
+                            }
+                        ]
+                    },
+                    "title": "profiles"
+                },
+                "name": {
+                    "type": "string",
+                    "title": "name"
+                },
+                "activeProfileIndex": {
+                    "type": "number",
+                    "title": "activeProfileIndex"
+                },
+                "exporter": {
+                    "type": "string",
+                    "title": "exporter"
+                }
+            },
+            "required": [
+                "$schema",
+                "profiles",
+                "shared"
+            ]
+        },
+        "FileFormat.Frame": {
+            "title": "FileFormat.Frame",
+            "type": "object",
+            "properties": {
+                "name": {
+                    "type": "string",
+                    "title": "name"
+                },
+                "file": {
+                    "type": "string",
+                    "title": "file"
+                },
+                "line": {
+                    "type": "number",
+                    "title": "line"
+                },
+                "col": {
+                    "type": "number",
+                    "title": "col"
+                }
+            },
+            "required": [
+                "name"
+            ]
+        },
+        "FileFormat.ProfileType": {
+            "title": "FileFormat.ProfileType",
+            "enum": [
+                "evented",
+                "sampled"
+            ],
+            "type": "string"
+        },
+        "FileFormat.IProfile": {
+            "title": "FileFormat.IProfile",
+            "type": "object",
+            "properties": {
+                "type": {
+                    "$ref": "#/definitions/FileFormat.ProfileType",
+                    "title": "type"
+                }
+            },
+            "required": [
+                "type"
+            ]
+        },
+        "FileFormat.EventedProfile": {
+            "title": "FileFormat.EventedProfile",
+            "type": "object",
+            "properties": {
+                "type": {
+                    "type": "string",
+                    "enum": [
+                        "evented"
+                    ],
+                    "title": "type"
+                },
+                "name": {
+                    "type": "string",
+                    "title": "name"
+                },
+                "unit": {
+                    "$ref": "#/definitions/FileFormat.ValueUnit",
+                    "title": "unit"
+                },
+                "startValue": {
+                    "type": "number",
+                    "title": "startValue"
+                },
+                "endValue": {
+                    "type": "number",
+                    "title": "endValue"
+                },
+                "events": {
+                    "type": "array",
+                    "items": {
+                        "anyOf": [
+                            {
+                                "$ref": "#/definitions/OpenFrameEvent"
+                            },
+                            {
+                                "$ref": "#/definitions/CloseFrameEvent"
+                            }
+                        ]
+                    },
+                    "title": "events"
+                }
+            },
+            "required": [
+                "endValue",
+                "events",
+                "name",
+                "startValue",
+                "type",
+                "unit"
+            ]
+        },
+        "SampledStack": {
+            "type": "array",
+            "items": {
+                "type": "number"
+            }
+        },
+        "FileFormat.SampledProfile": {
+            "title": "FileFormat.SampledProfile",
+            "type": "object",
+            "properties": {
+                "type": {
+                    "type": "string",
+                    "enum": [
+                        "sampled"
+                    ],
+                    "title": "type"
+                },
+                "name": {
+                    "type": "string",
+                    "title": "name"
+                },
+                "unit": {
+                    "$ref": "#/definitions/FileFormat.ValueUnit",
+                    "title": "unit"
+                },
+                "startValue": {
+                    "type": "number",
+                    "title": "startValue"
+                },
+                "endValue": {
+                    "type": "number",
+                    "title": "endValue"
+                },
+                "samples": {
+                    "type": "array",
+                    "items": {
+                        "type": "array",
+                        "items": {
+                            "type": "number"
+                        }
+                    },
+                    "title": "samples"
+                },
+                "weights": {
+                    "type": "array",
+                    "items": {
+                        "type": "number"
+                    },
+                    "title": "weights"
+                }
+            },
+            "required": [
+                "endValue",
+                "name",
+                "samples",
+                "startValue",
+                "type",
+                "unit",
+                "weights"
+            ]
+        },
+        "FileFormat.ValueUnit": {
+            "title": "FileFormat.ValueUnit",
+            "enum": [
+                "bytes",
+                "microseconds",
+                "milliseconds",
+                "nanoseconds",
+                "none",
+                "seconds"
+            ],
+            "type": "string"
+        },
+        "FileFormat.EventType": {
+            "title": "FileFormat.EventType",
+            "enum": [
+                "C",
+                "O"
+            ],
+            "type": "string"
+        },
+        "IEvent": {
+            "title": "IEvent",
+            "type": "object",
+            "properties": {
+                "type": {
+                    "$ref": "#/definitions/FileFormat.EventType",
+                    "title": "type"
+                },
+                "at": {
+                    "type": "number",
+                    "title": "at"
+                }
+            },
+            "required": [
+                "at",
+                "type"
+            ]
+        },
+        "OpenFrameEvent": {
+            "title": "OpenFrameEvent",
+            "type": "object",
+            "properties": {
+                "type": {
+                    "type": "string",
+                    "enum": [
+                        "O"
+                    ],
+                    "title": "type"
+                },
+                "frame": {
+                    "type": "number",
+                    "title": "frame"
+                },
+                "at": {
+                    "type": "number",
+                    "title": "at"
+                }
+            },
+            "required": [
+                "at",
+                "frame",
+                "type"
+            ]
+        },
+        "CloseFrameEvent": {
+            "title": "CloseFrameEvent",
+            "type": "object",
+            "properties": {
+                "type": {
+                    "type": "string",
+                    "enum": [
+                        "C"
+                    ],
+                    "title": "type"
+                },
+                "frame": {
+                    "type": "number",
+                    "title": "frame"
+                },
+                "at": {
+                    "type": "number",
+                    "title": "at"
+                }
+            },
+            "required": [
+                "at",
+                "frame",
+                "type"
+            ]
+        }
+    },
+    "$ref": "#/definitions/FileFormat.File"
+}

.venv/lib/python3.11/site-packages/ray/dashboard/client/build/speedscope-1.5.3/index.html

@@ -0,0 +1 @@
+<!DOCTYPE html><html lang="en"> <head> <meta charset="UTF-8"> <meta name="viewport" content="width=device-width, initial-scale=1.0"> <meta http-equiv="X-UA-Compatible" content="ie=edge"> <title>speedscope</title> <link href="https://fonts.googleapis.com/css?family=Source+Code+Pro" rel="stylesheet"> <script></script> <link rel="stylesheet" href="reset.7ae984ff.css"> <link rel="icon" type="image/png" sizes="32x32" href="favicon-32x32.1165a94e.png"> <link rel="icon" type="image/png" sizes="16x16" href="favicon-16x16.361d2b26.png"> </head> <body> <script src="speedscope.75eb7d8e.js"></script> </body> </html>

.venv/lib/python3.11/site-packages/ray/dashboard/client/build/speedscope-1.5.3/reset.7ae984ff.css

@@ -0,0 +1 @@
+a,abbr,acronym,address,applet,article,aside,audio,b,big,blockquote,body,canvas,caption,center,cite,code,dd,del,details,dfn,div,dl,dt,em,embed,fieldset,figcaption,figure,footer,form,h1,h2,h3,h4,h5,h6,header,hgroup,html,i,iframe,img,ins,kbd,label,legend,li,mark,menu,nav,object,ol,output,p,pre,q,ruby,s,samp,section,small,span,strike,strong,sub,summary,sup,table,tbody,td,tfoot,th,thead,time,tr,tt,u,ul,var,video{margin:0;padding:0;border:0;font-size:100%;font:inherit;vertical-align:baseline}article,aside,details,figcaption,figure,footer,header,hgroup,menu,nav,section{display:block}body{line-height:1}ol,ul{list-style:none}blockquote,q{quotes:none}blockquote:after,blockquote:before,q:after,q:before{content:"";content:none}table{border-collapse:collapse;border-spacing:0}html{overflow:hidden}body,html{height:100%}body{overflow:auto}

.venv/lib/python3.11/site-packages/ray/dashboard/client/build/static/css/main.388a904b.css

@@ -0,0 +1,2 @@
+@font-face{font-display:swap;font-family:Roboto;font-style:normal;font-weight:100;src:local("Roboto Thin "),local("Roboto-Thin"),url(../../static/media/roboto-latin-100.c2aa4ab115bf9c6057cb.woff2) format("woff2"),url(../../static/media/roboto-latin-100.a45108d3b34af91f9113.woff) format("woff")}@font-face{font-display:swap;font-family:Roboto;font-style:italic;font-weight:100;src:local("Roboto Thin italic"),local("Roboto-Thinitalic"),url(../../static/media/roboto-latin-100italic.7f839a8652da29745ce4.woff2) format("woff2"),url(../../static/media/roboto-latin-100italic.451d4e559d6f57cdf6a1.woff) format("woff")}@font-face{font-display:swap;font-family:Roboto;font-style:normal;font-weight:300;src:local("Roboto Light "),local("Roboto-Light"),url(../../static/media/roboto-latin-300.37a7069dc30fc663c878.woff2) format("woff2"),url(../../static/media/roboto-latin-300.865f928cbabcc9f8f2b5.woff) format("woff")}@font-face{font-display:swap;font-family:Roboto;font-style:italic;font-weight:300;src:local("Roboto Light italic"),local("Roboto-Lightitalic"),url(../../static/media/roboto-latin-300italic.c64e7e354c88e613c77c.woff2) format("woff2"),url(../../static/media/roboto-latin-300italic.bd5b7a13f2c52b531a2a.woff) format("woff")}@font-face{font-display:swap;font-family:Roboto;font-style:normal;font-weight:400;src:local("Roboto Regular "),local("Roboto-Regular"),url(../../static/media/roboto-latin-400.176f8f5bd5f02b3abfcf.woff2) format("woff2"),url(../../static/media/roboto-latin-400.49ae34d4cc6b98c00c69.woff) format("woff")}@font-face{font-display:swap;font-family:Roboto;font-style:italic;font-weight:400;src:local("Roboto Regular italic"),local("Roboto-Regularitalic"),url(../../static/media/roboto-latin-400italic.d022bc70dc1bf7b3425d.woff2) format("woff2"),url(../../static/media/roboto-latin-400italic.b1d9d9904bfca8802a63.woff) format("woff")}@font-face{font-display:swap;font-family:Roboto;font-style:normal;font-weight:500;src:local("Roboto Medium "),local("Roboto-Medium"),url(../../static/media/roboto-latin-500.f5b74d7ffcdf85b9dd60.woff2) format("woff2"),url(../../static/media/roboto-latin-500.cea99d3e3e13a3a599a0.woff) format("woff")}@font-face{font-display:swap;font-family:Roboto;font-style:italic;font-weight:500;src:local("Roboto Medium italic"),local("Roboto-Mediumitalic"),url(../../static/media/roboto-latin-500italic.0d8bb5b3ee5f5dac9e44.woff2) format("woff2"),url(../../static/media/roboto-latin-500italic.18d00f739ff1e1c52db1.woff) format("woff")}@font-face{font-display:swap;font-family:Roboto;font-style:normal;font-weight:700;src:local("Roboto Bold "),local("Roboto-Bold"),url(../../static/media/roboto-latin-700.c18ee39fb002ad58b6dc.woff2) format("woff2"),url(../../static/media/roboto-latin-700.2267169ee7270a22a963.woff) format("woff")}@font-face{font-display:swap;font-family:Roboto;font-style:italic;font-weight:700;src:local("Roboto Bold italic"),local("Roboto-Bolditalic"),url(../../static/media/roboto-latin-700italic.7d8125ff7f707231fd89.woff2) format("woff2"),url(../../static/media/roboto-latin-700italic.9360531f9bb817f917f0.woff) format("woff")}@font-face{font-display:swap;font-family:Roboto;font-style:normal;font-weight:900;src:local("Roboto Black "),local("Roboto-Black"),url(../../static/media/roboto-latin-900.870c8c1486f76054301a.woff2) format("woff2"),url(../../static/media/roboto-latin-900.bac8362e7a6ea60b6983.woff) format("woff")}@font-face{font-display:swap;font-family:Roboto;font-style:italic;font-weight:900;src:local("Roboto Black italic"),local("Roboto-Blackitalic"),url(../../static/media/roboto-latin-900italic.cb5ad999740e9d8a8bd1.woff2) format("woff2"),url(../../static/media/roboto-latin-900italic.c20d916c1a1b094c1cec.woff) format("woff")}.hljs-dark{color:#f8f8f2;display:block;overflow-x:auto;padding:.5em}.hljs-dark .hljs-keyword,.hljs-dark .hljs-link,.hljs-dark .hljs-literal,.hljs-dark .hljs-number,.hljs-dark .hljs-section,.hljs-dark .hljs-selector-tag{color:#8be9fd}.hljs-dark .hljs-function .hljs-keyword{color:#ff79c6}.hljs-dark .hljs-addition,.hljs-dark .hljs-attribute,.hljs-dark .hljs-bullet,.hljs-dark .hljs-name,.hljs-dark .hljs-string,.hljs-dark .hljs-symbol,.hljs-dark .hljs-template-tag,.hljs-dark .hljs-template-variable,.hljs-dark .hljs-title,.hljs-dark .hljs-type,.hljs-dark .hljs-variable{color:#f1fa8c}.hljs-dark .hljs-comment,.hljs-dark .hljs-deletion,.hljs-dark .hljs-meta,.hljs-dark .hljs-quote{color:#6272a4}.hljs-dark .hljs-doctag,.hljs-dark .hljs-keyword,.hljs-dark .hljs-literal,.hljs-dark .hljs-name,.hljs-dark .hljs-section,.hljs-dark .hljs-selector-tag,.hljs-dark .hljs-strong,.hljs-dark .hljs-title,.hljs-dark .hljs-type{font-weight:700}.hljs-dark .hljs-emphasis{font-style:italic}.hljs-light{color:#333;display:block;overflow-x:auto;padding:.5em}.hljs-light .hljs-comment,.hljs-light .hljs-quote{color:#998;font-style:italic}.hljs-light .hljs-keyword,.hljs-light .hljs-selector-tag,.hljs-light .hljs-subst{color:#333;font-weight:700}.hljs-light .hljs-literal,.hljs-light .hljs-number,.hljs-light .hljs-tag .hljs-attr,.hljs-light .hljs-template-variable,.hljs-light .hljs-variable{color:teal}.hljs-light .hljs-doctag,.hljs-light .hljs-string{color:#d14}.hljs-light .hljs-section,.hljs-light .hljs-selector-id,.hljs-light .hljs-title{color:#900;font-weight:700}.hljs-light .hljs-subst{font-weight:400}.hljs-light .hljs-class .hljs-title,.hljs-light .hljs-type{color:#458;font-weight:700}.hljs-light .hljs-attribute,.hljs-light .hljs-name,.hljs-light .hljs-tag{color:navy;font-weight:400}.hljs-light .hljs-link,.hljs-light .hljs-regexp{color:#009926}.hljs-light .hljs-bullet,.hljs-light .hljs-symbol{color:#990073}.hljs-light .hljs-built_in,.hljs-light .hljs-builtin-name{color:#0086b3}.hljs-light .hljs-meta{color:#999;font-weight:700}.hljs-light .hljs-deletion{background:#fdd}.hljs-light .hljs-addition{background:#dfd}.hljs-light .hljs-emphasis{font-style:italic}.hljs-light .hljs-strong{font-weight:700}span.find-kws{background-color:#ffd800}
+/*# sourceMappingURL=main.388a904b.css.map*/

.venv/lib/python3.11/site-packages/ray/experimental/__init__.py

@@ -0,0 +1,10 @@
+from ray.experimental.dynamic_resources import set_resource
+from ray.experimental.locations import get_local_object_locations, get_object_locations
+from ray.experimental.packaging.load_package import load_package
+
+__all__ = [
+    "get_object_locations",
+    "get_local_object_locations",
+    "set_resource",
+    "load_package",
+]

.venv/lib/python3.11/site-packages/ray/experimental/array/distributed/__init__.py

@@ -0,0 +1,40 @@
+from . import linalg, random
+from .core import (
+    BLOCK_SIZE,
+    DistArray,
+    add,
+    assemble,
+    blockwise_dot,
+    copy,
+    dot,
+    eye,
+    numpy_to_dist,
+    ones,
+    subblocks,
+    subtract,
+    transpose,
+    tril,
+    triu,
+    zeros,
+)
+
+__all__ = [
+    "random",
+    "linalg",
+    "BLOCK_SIZE",
+    "DistArray",
+    "assemble",
+    "zeros",
+    "ones",
+    "copy",
+    "eye",
+    "triu",
+    "tril",
+    "blockwise_dot",
+    "dot",
+    "transpose",
+    "add",
+    "subtract",
+    "numpy_to_dist",
+    "subblocks",
+]

.venv/lib/python3.11/site-packages/ray/experimental/array/distributed/core.py

@@ -0,0 +1,316 @@
+import numpy as np
+
+import ray
+import ray.experimental.array.remote as ra
+
+BLOCK_SIZE = 10
+
+
+class DistArray:
+    def __init__(self, shape, object_refs=None):
+        self.shape = shape
+        self.ndim = len(shape)
+        self.num_blocks = [int(np.ceil(1.0 * a / BLOCK_SIZE)) for a in self.shape]
+        if object_refs is not None:
+            self.object_refs = object_refs
+        else:
+            self.object_refs = np.empty(self.num_blocks, dtype=object)
+        if self.num_blocks != list(self.object_refs.shape):
+            raise Exception(
+                "The fields `num_blocks` and `object_refs` are "
+                "inconsistent, `num_blocks` is {} and `object_refs` "
+                "has shape {}".format(self.num_blocks, list(self.object_refs.shape))
+            )
+
+    @staticmethod
+    def compute_block_lower(index, shape):
+        if len(index) != len(shape):
+            raise Exception(
+                "The fields `index` and `shape` must have the "
+                "same length, but `index` is {} and `shape` is "
+                "{}.".format(index, shape)
+            )
+        return [elem * BLOCK_SIZE for elem in index]
+
+    @staticmethod
+    def compute_block_upper(index, shape):
+        if len(index) != len(shape):
+            raise Exception(
+                "The fields `index` and `shape` must have the "
+                "same length, but `index` is {} and `shape` is "
+                "{}.".format(index, shape)
+            )
+        upper = []
+        for i in range(len(shape)):
+            upper.append(min((index[i] + 1) * BLOCK_SIZE, shape[i]))
+        return upper
+
+    @staticmethod
+    def compute_block_shape(index, shape):
+        lower = DistArray.compute_block_lower(index, shape)
+        upper = DistArray.compute_block_upper(index, shape)
+        return [u - l for (l, u) in zip(lower, upper)]
+
+    @staticmethod
+    def compute_num_blocks(shape):
+        return [int(np.ceil(1.0 * a / BLOCK_SIZE)) for a in shape]
+
+    def assemble(self):
+        """Assemble an array from a distributed array of object refs."""
+        first_block = ray.get(self.object_refs[(0,) * self.ndim])
+        dtype = first_block.dtype
+        result = np.zeros(self.shape, dtype=dtype)
+        for index in np.ndindex(*self.num_blocks):
+            lower = DistArray.compute_block_lower(index, self.shape)
+            upper = DistArray.compute_block_upper(index, self.shape)
+            value = ray.get(self.object_refs[index])
+            result[tuple(slice(l, u) for (l, u) in zip(lower, upper))] = value
+        return result
+
+    def __getitem__(self, sliced):
+        # TODO(rkn): Fix this, this is just a placeholder that should work but
+        # is inefficient.
+        a = self.assemble()
+        return a[sliced]
+
+
+@ray.remote
+def assemble(a):
+    return a.assemble()
+
+
+# TODO(rkn): What should we call this method?
+@ray.remote
+def numpy_to_dist(a):
+    result = DistArray(a.shape)
+    for index in np.ndindex(*result.num_blocks):
+        lower = DistArray.compute_block_lower(index, a.shape)
+        upper = DistArray.compute_block_upper(index, a.shape)
+        idx = tuple(slice(l, u) for (l, u) in zip(lower, upper))
+        result.object_refs[index] = ray.put(a[idx])
+    return result
+
+
+@ray.remote
+def zeros(shape, dtype_name="float"):
+    result = DistArray(shape)
+    for index in np.ndindex(*result.num_blocks):
+        result.object_refs[index] = ra.zeros.remote(
+            DistArray.compute_block_shape(index, shape), dtype_name=dtype_name
+        )
+    return result
+
+
+@ray.remote
+def ones(shape, dtype_name="float"):
+    result = DistArray(shape)
+    for index in np.ndindex(*result.num_blocks):
+        result.object_refs[index] = ra.ones.remote(
+            DistArray.compute_block_shape(index, shape), dtype_name=dtype_name
+        )
+    return result
+
+
+@ray.remote
+def copy(a):
+    result = DistArray(a.shape)
+    for index in np.ndindex(*result.num_blocks):
+        # We don't need to actually copy the objects because remote objects are
+        # immutable.
+        result.object_refs[index] = a.object_refs[index]
+    return result
+
+
+@ray.remote
+def eye(dim1, dim2=-1, dtype_name="float"):
+    dim2 = dim1 if dim2 == -1 else dim2
+    shape = [dim1, dim2]
+    result = DistArray(shape)
+    for (i, j) in np.ndindex(*result.num_blocks):
+        block_shape = DistArray.compute_block_shape([i, j], shape)
+        if i == j:
+            result.object_refs[i, j] = ra.eye.remote(
+                block_shape[0], block_shape[1], dtype_name=dtype_name
+            )
+        else:
+            result.object_refs[i, j] = ra.zeros.remote(
+                block_shape, dtype_name=dtype_name
+            )
+    return result
+
+
+@ray.remote
+def triu(a):
+    if a.ndim != 2:
+        raise Exception(
+            "Input must have 2 dimensions, but a.ndim is {}.".format(a.ndim)
+        )
+    result = DistArray(a.shape)
+    for (i, j) in np.ndindex(*result.num_blocks):
+        if i < j:
+            result.object_refs[i, j] = ra.copy.remote(a.object_refs[i, j])
+        elif i == j:
+            result.object_refs[i, j] = ra.triu.remote(a.object_refs[i, j])
+        else:
+            result.object_refs[i, j] = ra.zeros_like.remote(a.object_refs[i, j])
+    return result
+
+
+@ray.remote
+def tril(a):
+    if a.ndim != 2:
+        raise Exception(
+            "Input must have 2 dimensions, but a.ndim is {}.".format(a.ndim)
+        )
+    result = DistArray(a.shape)
+    for (i, j) in np.ndindex(*result.num_blocks):
+        if i > j:
+            result.object_refs[i, j] = ra.copy.remote(a.object_refs[i, j])
+        elif i == j:
+            result.object_refs[i, j] = ra.tril.remote(a.object_refs[i, j])
+        else:
+            result.object_refs[i, j] = ra.zeros_like.remote(a.object_refs[i, j])
+    return result
+
+
+@ray.remote
+def blockwise_dot(*matrices):
+    n = len(matrices)
+    if n % 2 != 0:
+        raise Exception(
+            "blockwise_dot expects an even number of arguments, "
+            "but len(matrices) is {}.".format(n)
+        )
+    shape = (matrices[0].shape[0], matrices[n // 2].shape[1])
+    result = np.zeros(shape)
+    for i in range(n // 2):
+        result += np.dot(matrices[i], matrices[n // 2 + i])
+    return result
+
+
+@ray.remote
+def dot(a, b):
+    if a.ndim != 2:
+        raise Exception(
+            "dot expects its arguments to be 2-dimensional, but "
+            "a.ndim = {}.".format(a.ndim)
+        )
+    if b.ndim != 2:
+        raise Exception(
+            "dot expects its arguments to be 2-dimensional, but "
+            "b.ndim = {}.".format(b.ndim)
+        )
+    if a.shape[1] != b.shape[0]:
+        raise Exception(
+            "dot expects a.shape[1] to equal b.shape[0], but "
+            "a.shape = {} and b.shape = {}.".format(a.shape, b.shape)
+        )
+    shape = [a.shape[0], b.shape[1]]
+    result = DistArray(shape)
+    for (i, j) in np.ndindex(*result.num_blocks):
+        args = list(a.object_refs[i, :]) + list(b.object_refs[:, j])
+        result.object_refs[i, j] = blockwise_dot.remote(*args)
+    return result
+
+
+@ray.remote
+def subblocks(a, *ranges):
+    """
+    This function produces a distributed array from a subset of the blocks in
+    the `a`. The result and `a` will have the same number of dimensions. For
+    example,
+        subblocks(a, [0, 1], [2, 4])
+    will produce a DistArray whose object_refs are
+        [[a.object_refs[0, 2], a.object_refs[0, 4]],
+         [a.object_refs[1, 2], a.object_refs[1, 4]]]
+    We allow the user to pass in an empty list [] to indicate the full range.
+    """
+    ranges = list(ranges)
+    if len(ranges) != a.ndim:
+        raise Exception(
+            "sub_blocks expects to receive a number of ranges "
+            "equal to a.ndim, but it received {} ranges and "
+            "a.ndim = {}.".format(len(ranges), a.ndim)
+        )
+    for i in range(len(ranges)):
+        # We allow the user to pass in an empty list to indicate the full
+        # range.
+        if ranges[i] == []:
+            ranges[i] = range(a.num_blocks[i])
+        if not np.alltrue(ranges[i] == np.sort(ranges[i])):
+            raise Exception(
+                "Ranges passed to sub_blocks must be sorted, but "
+                "the {}th range is {}.".format(i, ranges[i])
+            )
+        if ranges[i][0] < 0:
+            raise Exception(
+                "Values in the ranges passed to sub_blocks must "
+                "be at least 0, but the {}th range is {}.".format(i, ranges[i])
+            )
+        if ranges[i][-1] >= a.num_blocks[i]:
+            raise Exception(
+                "Values in the ranges passed to sub_blocks must "
+                "be less than the relevant number of blocks, but "
+                "the {}th range is {}, and a.num_blocks = {}.".format(
+                    i, ranges[i], a.num_blocks
+                )
+            )
+    last_index = [r[-1] for r in ranges]
+    last_block_shape = DistArray.compute_block_shape(last_index, a.shape)
+    shape = [
+        (len(ranges[i]) - 1) * BLOCK_SIZE + last_block_shape[i] for i in range(a.ndim)
+    ]
+    result = DistArray(shape)
+    for index in np.ndindex(*result.num_blocks):
+        result.object_refs[index] = a.object_refs[
+            tuple(ranges[i][index[i]] for i in range(a.ndim))
+        ]
+    return result
+
+
+@ray.remote
+def transpose(a):
+    if a.ndim != 2:
+        raise Exception(
+            "transpose expects its argument to be 2-dimensional, "
+            "but a.ndim = {}, a.shape = {}.".format(a.ndim, a.shape)
+        )
+    result = DistArray([a.shape[1], a.shape[0]])
+    for i in range(result.num_blocks[0]):
+        for j in range(result.num_blocks[1]):
+            result.object_refs[i, j] = ra.transpose.remote(a.object_refs[j, i])
+    return result
+
+
+# TODO(rkn): support broadcasting?
+@ray.remote
+def add(x1, x2):
+    if x1.shape != x2.shape:
+        raise Exception(
+            "add expects arguments `x1` and `x2` to have the same "
+            "shape, but x1.shape = {}, and x2.shape = {}.".format(x1.shape, x2.shape)
+        )
+    result = DistArray(x1.shape)
+    for index in np.ndindex(*result.num_blocks):
+        result.object_refs[index] = ra.add.remote(
+            x1.object_refs[index], x2.object_refs[index]
+        )
+    return result
+
+
+# TODO(rkn): support broadcasting?
+@ray.remote
+def subtract(x1, x2):
+    if x1.shape != x2.shape:
+        raise Exception(
+            "subtract expects arguments `x1` and `x2` to have the "
+            "same shape, but x1.shape = {}, and x2.shape = {}.".format(
+                x1.shape, x2.shape
+            )
+        )
+    result = DistArray(x1.shape)
+    for index in np.ndindex(*result.num_blocks):
+        result.object_refs[index] = ra.subtract.remote(
+            x1.object_refs[index], x2.object_refs[index]
+        )
+    return result

.venv/lib/python3.11/site-packages/ray/experimental/array/distributed/linalg.py

@@ -0,0 +1,231 @@
+import numpy as np
+
+import ray
+import ray.experimental.array.remote as ra
+from . import core
+
+__all__ = ["tsqr", "modified_lu", "tsqr_hr", "qr"]
+
+
+@ray.remote(num_returns=2)
+def tsqr(a):
+    """Perform a QR decomposition of a tall-skinny matrix.
+
+    Args:
+        a: A distributed matrix with shape MxN (suppose K = min(M, N)).
+
+    Returns:
+        A tuple of q (a DistArray) and r (a numpy array) satisfying the
+            following.
+            - If q_full = ray.get(DistArray, q).assemble(), then
+              q_full.shape == (M, K).
+            - np.allclose(np.dot(q_full.T, q_full), np.eye(K)) == True.
+            - If r_val = ray.get(np.ndarray, r), then r_val.shape == (K, N).
+            - np.allclose(r, np.triu(r)) == True.
+    """
+    if len(a.shape) != 2:
+        raise Exception(
+            "tsqr requires len(a.shape) == 2, but a.shape is {}".format(a.shape)
+        )
+    if a.num_blocks[1] != 1:
+        raise Exception(
+            "tsqr requires a.num_blocks[1] == 1, but a.num_blocks "
+            "is {}".format(a.num_blocks)
+        )
+
+    num_blocks = a.num_blocks[0]
+    K = int(np.ceil(np.log2(num_blocks))) + 1
+    q_tree = np.empty((num_blocks, K), dtype=object)
+    current_rs = []
+    for i in range(num_blocks):
+        block = a.object_refs[i, 0]
+        q, r = ra.linalg.qr.remote(block)
+        q_tree[i, 0] = q
+        current_rs.append(r)
+    for j in range(1, K):
+        new_rs = []
+        for i in range(int(np.ceil(1.0 * len(current_rs) / 2))):
+            stacked_rs = ra.vstack.remote(*current_rs[(2 * i) : (2 * i + 2)])
+            q, r = ra.linalg.qr.remote(stacked_rs)
+            q_tree[i, j] = q
+            new_rs.append(r)
+        current_rs = new_rs
+    assert len(current_rs) == 1, "len(current_rs) = " + str(len(current_rs))
+
+    # handle the special case in which the whole DistArray "a" fits in one
+    # block and has fewer rows than columns, this is a bit ugly so think about
+    # how to remove it
+    if a.shape[0] >= a.shape[1]:
+        q_shape = a.shape
+    else:
+        q_shape = [a.shape[0], a.shape[0]]
+    q_num_blocks = core.DistArray.compute_num_blocks(q_shape)
+    q_object_refs = np.empty(q_num_blocks, dtype=object)
+    q_result = core.DistArray(q_shape, q_object_refs)
+
+    # reconstruct output
+    for i in range(num_blocks):
+        q_block_current = q_tree[i, 0]
+        ith_index = i
+        for j in range(1, K):
+            if np.mod(ith_index, 2) == 0:
+                lower = [0, 0]
+                upper = [a.shape[1], core.BLOCK_SIZE]
+            else:
+                lower = [a.shape[1], 0]
+                upper = [2 * a.shape[1], core.BLOCK_SIZE]
+            ith_index //= 2
+            q_block_current = ra.dot.remote(
+                q_block_current, ra.subarray.remote(q_tree[ith_index, j], lower, upper)
+            )
+        q_result.object_refs[i] = q_block_current
+    r = current_rs[0]
+    return q_result, ray.get(r)
+
+
+# TODO(rkn): This is unoptimized, we really want a block version of this.
+# This is Algorithm 5 from
+# http://www.eecs.berkeley.edu/Pubs/TechRpts/2013/EECS-2013-175.pdf.
+@ray.remote(num_returns=3)
+def modified_lu(q):
+    """Perform a modified LU decomposition of a matrix.
+
+    This takes a matrix q with orthonormal columns, returns l, u, s such that
+    q - s = l * u.
+
+    Args:
+        q: A two dimensional orthonormal matrix q.
+
+    Returns:
+        A tuple of a lower triangular matrix l, an upper triangular matrix u,
+            and a a vector representing a diagonal matrix s such that
+            q - s = l * u.
+    """
+    q = q.assemble()
+    m, b = q.shape[0], q.shape[1]
+    S = np.zeros(b)
+
+    q_work = np.copy(q)
+
+    for i in range(b):
+        S[i] = -1 * np.sign(q_work[i, i])
+        q_work[i, i] -= S[i]
+        # Scale ith column of L by diagonal element.
+        q_work[(i + 1) : m, i] /= q_work[i, i]
+        # Perform Schur complement update.
+        q_work[(i + 1) : m, (i + 1) : b] -= np.outer(
+            q_work[(i + 1) : m, i], q_work[i, (i + 1) : b]
+        )
+
+    L = np.tril(q_work)
+    for i in range(b):
+        L[i, i] = 1
+    U = np.triu(q_work)[:b, :]
+    # TODO(rkn): Get rid of the put below.
+    return ray.get(core.numpy_to_dist.remote(ray.put(L))), U, S
+
+
+@ray.remote(num_returns=2)
+def tsqr_hr_helper1(u, s, y_top_block, b):
+    y_top = y_top_block[:b, :b]
+    s_full = np.diag(s)
+    t = -1 * np.dot(u, np.dot(s_full, np.linalg.inv(y_top).T))
+    return t, y_top
+
+
+@ray.remote
+def tsqr_hr_helper2(s, r_temp):
+    s_full = np.diag(s)
+    return np.dot(s_full, r_temp)
+
+
+# This is Algorithm 6 from
+# http://www.eecs.berkeley.edu/Pubs/TechRpts/2013/EECS-2013-175.pdf.
+@ray.remote(num_returns=4)
+def tsqr_hr(a):
+    q, r_temp = tsqr.remote(a)
+    y, u, s = modified_lu.remote(q)
+    y_blocked = ray.get(y)
+    t, y_top = tsqr_hr_helper1.remote(u, s, y_blocked.object_refs[0, 0], a.shape[1])
+    r = tsqr_hr_helper2.remote(s, r_temp)
+    return ray.get(y), ray.get(t), ray.get(y_top), ray.get(r)
+
+
+@ray.remote
+def qr_helper1(a_rc, y_ri, t, W_c):
+    return a_rc - np.dot(y_ri, np.dot(t.T, W_c))
+
+
+@ray.remote
+def qr_helper2(y_ri, a_rc):
+    return np.dot(y_ri.T, a_rc)
+
+
+# This is Algorithm 7 from
+# http://www.eecs.berkeley.edu/Pubs/TechRpts/2013/EECS-2013-175.pdf.
+@ray.remote(num_returns=2)
+def qr(a):
+
+    m, n = a.shape[0], a.shape[1]
+    k = min(m, n)
+
+    # we will store our scratch work in a_work
+    a_work = core.DistArray(a.shape, np.copy(a.object_refs))
+
+    result_dtype = np.linalg.qr(ray.get(a.object_refs[0, 0]))[0].dtype.name
+    # TODO(rkn): It would be preferable not to get this right after creating
+    # it.
+    r_res = ray.get(core.zeros.remote([k, n], result_dtype))
+    # TODO(rkn): It would be preferable not to get this right after creating
+    # it.
+    y_res = ray.get(core.zeros.remote([m, k], result_dtype))
+    Ts = []
+
+    # The for loop differs from the paper, which says
+    # "for i in range(a.num_blocks[1])", but that doesn't seem to make any
+    # sense when a.num_blocks[1] > a.num_blocks[0].
+    for i in range(min(a.num_blocks[0], a.num_blocks[1])):
+        sub_dist_array = core.subblocks.remote(
+            a_work, list(range(i, a_work.num_blocks[0])), [i]
+        )
+        y, t, _, R = tsqr_hr.remote(sub_dist_array)
+        y_val = ray.get(y)
+
+        for j in range(i, a.num_blocks[0]):
+            y_res.object_refs[j, i] = y_val.object_refs[j - i, 0]
+        if a.shape[0] > a.shape[1]:
+            # in this case, R needs to be square
+            R_shape = ray.get(ra.shape.remote(R))
+            eye_temp = ra.eye.remote(R_shape[1], R_shape[0], dtype_name=result_dtype)
+            r_res.object_refs[i, i] = ra.dot.remote(eye_temp, R)
+        else:
+            r_res.object_refs[i, i] = R
+        Ts.append(core.numpy_to_dist.remote(t))
+
+        for c in range(i + 1, a.num_blocks[1]):
+            W_rcs = []
+            for r in range(i, a.num_blocks[0]):
+                y_ri = y_val.object_refs[r - i, 0]
+                W_rcs.append(qr_helper2.remote(y_ri, a_work.object_refs[r, c]))
+            W_c = ra.sum_list.remote(*W_rcs)
+            for r in range(i, a.num_blocks[0]):
+                y_ri = y_val.object_refs[r - i, 0]
+                A_rc = qr_helper1.remote(a_work.object_refs[r, c], y_ri, t, W_c)
+                a_work.object_refs[r, c] = A_rc
+            r_res.object_refs[i, c] = a_work.object_refs[i, c]
+
+    # construct q_res from Ys and Ts
+    q = core.eye.remote(m, k, dtype_name=result_dtype)
+    for i in range(len(Ts))[::-1]:
+        y_col_block = core.subblocks.remote(y_res, [], [i])
+        q = core.subtract.remote(
+            q,
+            core.dot.remote(
+                y_col_block,
+                core.dot.remote(
+                    Ts[i], core.dot.remote(core.transpose.remote(y_col_block), q)
+                ),
+            ),
+        )
+
+    return ray.get(q), r_res

.venv/lib/python3.11/site-packages/ray/experimental/array/distributed/random.py

@@ -0,0 +1,17 @@
+import numpy as np
+
+import ray
+import ray.experimental.array.remote as ra
+from .core import DistArray
+
+
+@ray.remote
+def normal(shape):
+    num_blocks = DistArray.compute_num_blocks(shape)
+    object_refs = np.empty(num_blocks, dtype=object)
+    for index in np.ndindex(*num_blocks):
+        object_refs[index] = ra.random.normal.remote(
+            DistArray.compute_block_shape(index, shape)
+        )
+    result = DistArray(shape, object_refs)
+    return result

.venv/lib/python3.11/site-packages/ray/experimental/channel/auto_transport_type.py

@@ -0,0 +1,167 @@
+from typing import Dict, List, Optional, Tuple, Union
+
+import ray
+from ray.experimental.channel import ChannelOutputType
+from ray.experimental.channel.torch_tensor_type import TorchTensorType
+
+
+class AutoTransportType(ChannelOutputType):
+    """
+    Type hint for automatic transport selection for tensors.
+
+    With this type hint Compiled Graphs automatically decide the best transport
+    to use (e.g., NCCL or shared memory) based on the node locations and GPU IDs
+    of the readers and writers.
+    """
+
+    def __init__(self, _static_shape: bool = False, _direct_return: bool = False):
+        self._static_shape = _static_shape
+        self._direct_return = _direct_return
+
+    def create_channel(
+        self,
+        writer: Optional["ray.actor.ActorHandle"],
+        reader_and_node_list: List[Tuple["ray.actor.ActorHandle", str]],
+        driver_actor_id: Optional[str] = None,
+    ) -> "ChannelOutputType":
+        """
+        Directly calling create_channel() on AutoTransportType should not happen,
+        just raise an exception with informative message.
+        """
+        raise ValueError(
+            "This should not happen: AutoTransportType should "
+            "have been resolved before creating a channel. "
+            "Please file a Ray GitHub issue for bug report."
+        )
+
+
+class TypeHintResolver:
+    """
+    This class is used to resolve `AutoChannelType` into an actual channel type
+    (e.g., `TorchTensorType` with proper transport) based on node locations and
+    GPU IDs of the readers and writers.
+    """
+
+    def __init__(self, actor_to_gpu_ids: Dict["ray.actor.ActorHandle", List[str]]):
+        """
+        Args:
+            actor_to_gpu_ids: Mapping from actor handle to its GPU IDs.
+        """
+        self._actor_to_gpu_ids = actor_to_gpu_ids
+
+    def _get_gpu_ids(self, actor: "ray.actor.ActorHandle") -> List[str]:
+        """
+        Get the GPU IDs of the actor.
+
+        Returns:
+            The GPU IDs of the actor. If the actor is not found,
+            return an empty list.
+        """
+        gpu_ids = self._actor_to_gpu_ids.get(actor, [])
+        assert len(gpu_ids) <= 1, (
+            "Compiled Graphs currently don't support allocating multiple GPUs "
+            "to a single actor"
+        )
+        return gpu_ids
+
+    def _use_same_gpu(
+        self,
+        writer_and_node: Tuple["ray.actor.ActorHandle", str],
+        reader_and_node: Union[
+            Tuple["ray.actor.ActorHandle", str],
+            List[Tuple["ray.actor.ActorHandle", str]],
+        ],
+    ) -> bool:
+        """
+        Check if the writer and readers use the same GPU.
+
+        Args:
+            writer_and_node: A tuple of writer actor handle and its node ID.
+            reader_and_node: A tuple of reader actor handle and its node ID, or
+                a list of such tuples.
+
+        Returns:
+            True if the writer and all the readers use the same GPU, False otherwise.
+        """
+        if isinstance(reader_and_node, list):
+            return all(
+                self._use_same_gpu(writer_and_node, entry) for entry in reader_and_node
+            )
+        if writer_and_node[1] != reader_and_node[1]:
+            return False
+        writer_gpu_ids = self._get_gpu_ids(writer_and_node[0])
+        reader_gpu_ids = self._get_gpu_ids(reader_and_node[0])
+        return writer_gpu_ids == reader_gpu_ids
+
+    def _use_gpu(
+        self, actors: Union["ray.actor.ActorHandle", List["ray.actor.ActorHandle"]]
+    ) -> bool:
+        """
+        Check if the actors use GPUs.
+
+        Args:
+            actors: An actor handle or a list of actor handles.
+
+        Returns:
+            True if the actors use GPUs, False otherwise.
+        """
+        if isinstance(actors, list):
+            return all(self._use_gpu(actor) for actor in actors)
+        gpu_ids = self._get_gpu_ids(actors)
+        return len(gpu_ids) > 0
+
+    def resolve(
+        self,
+        auto_transport_type: AutoTransportType,
+        writer_and_node: Tuple[Optional["ray.actor.ActorHandle"], str],
+        reader_and_node_list: List[Tuple[Optional["ray.actor.ActorHandle"], str]],
+    ) -> "ChannelOutputType":
+        """
+        Resolve auto_transport_type to the actual channel output type
+        based on the node locations and GPU IDs.
+
+        Args:
+            auto_transport_type: The type to resolve
+            writer_and_node: A tuple of writer actor handle and its node ID.
+                A None writer actor handle means the writer is the driver.
+            reader_and_node_list: A list of tuples of reader actor handle and its
+                node ID. A None reader actor handle means the reader is the driver.
+
+        Returns:
+            The actual channel type.
+        """
+        writer = writer_and_node[0]
+        readers = [reader for reader, _ in reader_and_node_list]
+
+        if writer is None or any(reader is None for reader in readers):
+            # None means actor is the driver, currently driver on GPU
+            # is not supported, so we always use shared memory to transfer
+            # tensors.
+            return TorchTensorType(
+                _static_shape=auto_transport_type._static_shape,
+                _direct_return=auto_transport_type._direct_return,
+            )
+
+        # Case 1: writer and readers don't both use GPU, use shared memory
+        # to transport the tensors
+        if not (self._use_gpu(writer) and self._use_gpu(readers)):
+            return TorchTensorType(
+                _static_shape=auto_transport_type._static_shape,
+                _direct_return=auto_transport_type._direct_return,
+            )
+
+        # Case 2: writer and readers use the same GPU are are on the same node,
+        # use shared memory to transport the tensors
+        if self._use_same_gpu(writer_and_node, reader_and_node_list):
+            return TorchTensorType(
+                _static_shape=auto_transport_type._static_shape,
+                _direct_return=auto_transport_type._direct_return,
+            )
+
+        # Case 3: writer and readers use different GPUs, use NCCL to transport
+        # the tensors
+        return TorchTensorType(
+            transport="nccl",
+            _static_shape=auto_transport_type._static_shape,
+            _direct_return=auto_transport_type._direct_return,
+        )

.venv/lib/python3.11/site-packages/ray/experimental/channel/cached_channel.py

@@ -0,0 +1,111 @@
+import uuid
+from typing import Any, Optional
+
+from ray.experimental.channel.common import ChannelInterface
+
+
+class CachedChannel(ChannelInterface):
+    """
+    CachedChannel wraps an inner channel and caches the data read from it until
+    `num_reads` reads have completed. If inner channel is None, the data
+    is written to serialization context and retrieved from there. This is useful
+    when passing data within the same actor and a shared memory channel can be
+    avoided.
+
+    Args:
+        num_reads: The number of reads from this channel that must happen before
+            writing again. Readers must be methods of the same actor.
+        inner_channel: The inner channel to cache data from. If None, the data is
+            read from the serialization context.
+        _channel_id: The unique ID for the channel. If None, a new ID is generated.
+    """
+
+    def __init__(
+        self,
+        num_reads: int,
+        inner_channel: Optional[ChannelInterface] = None,
+        _channel_id: Optional[str] = None,
+    ):
+        assert num_reads > 0, "num_reads must be greater than 0."
+        self._num_reads = num_reads
+        self._inner_channel = inner_channel
+        # Generate a unique ID for the channel. The writer and reader will use
+        # this ID to store and retrieve data from the _SerializationContext.
+        self._channel_id = _channel_id
+        if self._channel_id is None:
+            self._channel_id = str(uuid.uuid4())
+
+    def ensure_registered_as_writer(self) -> None:
+        if self._inner_channel is not None:
+            self._inner_channel.ensure_registered_as_writer()
+
+    def ensure_registered_as_reader(self) -> None:
+        if self._inner_channel is not None:
+            self._inner_channel.ensure_registered_as_reader()
+
+    def __reduce__(self):
+        return CachedChannel, (
+            self._num_reads,
+            self._inner_channel,
+            self._channel_id,
+        )
+
+    def __str__(self) -> str:
+        return (
+            f"CachedChannel(channel_id={self._channel_id}, "
+            f"num_reads={self._num_reads}), "
+            f"inner_channel={self._inner_channel})"
+        )
+
+    def write(self, value: Any, timeout: Optional[float] = None):
+        self.ensure_registered_as_writer()
+        # TODO: better organize the imports
+        from ray.experimental.channel import ChannelContext
+
+        if self._inner_channel is not None:
+            self._inner_channel.write(value, timeout)
+            return
+
+        # Otherwise no need to check timeout as the operation is non-blocking.
+
+        # Because both the reader and writer are in the same worker process,
+        # we can directly store the data in the context instead of storing
+        # it in the channel object. This removes the serialization overhead of `value`.
+        ctx = ChannelContext.get_current().serialization_context
+        ctx.set_data(self._channel_id, value, self._num_reads)
+
+    def read(self, timeout: Optional[float] = None) -> Any:
+        self.ensure_registered_as_reader()
+        # TODO: better organize the imports
+        from ray.experimental.channel import ChannelContext
+
+        ctx = ChannelContext.get_current().serialization_context
+        if ctx.has_data(self._channel_id):
+            # No need to check timeout as the operation is non-blocking.
+            return ctx.get_data(self._channel_id)
+
+        assert (
+            self._inner_channel is not None
+        ), "Cannot read from the serialization context while inner channel is None."
+        value = self._inner_channel.read(timeout)
+        ctx.set_data(self._channel_id, value, self._num_reads)
+        # NOTE: Currently we make a contract with Compiled Graph users that the
+        # channel results should not be mutated by the actor methods.
+        # When the user needs to modify the channel results, they should
+        # make a copy of the channel results and modify the copy.
+        # This is the same contract as used in IntraProcessChannel.
+        # This contract is NOT enforced right now in either case.
+        # TODO(rui): introduce a flag to control the behavior:
+        # for example, by default we make a deep copy of the channel
+        # result, but the user can turn off the deep copy for performance
+        # improvements.
+        # https://github.com/ray-project/ray/issues/47409
+        return ctx.get_data(self._channel_id)
+
+    def close(self) -> None:
+        from ray.experimental.channel import ChannelContext
+
+        if self._inner_channel is not None:
+            self._inner_channel.close()
+        ctx = ChannelContext.get_current().serialization_context
+        ctx.reset_data(self._channel_id)

.venv/lib/python3.11/site-packages/ray/experimental/channel/intra_process_channel.py

@@ -0,0 +1,70 @@
+import uuid
+from typing import Any, Optional
+
+from ray.experimental.channel import ChannelContext
+from ray.experimental.channel.common import ChannelInterface
+from ray.util.annotations import PublicAPI
+
+
+@PublicAPI(stability="alpha")
+class IntraProcessChannel(ChannelInterface):
+    """
+    IntraProcessChannel is a channel for communication between two tasks in the same
+    worker process. It writes data directly to the worker's _SerializationContext
+    and reads data from the _SerializationContext to avoid the serialization
+    overhead and the need for reading/writing from shared memory. Note that if the
+    readers may mutate the data, users should deep copy the data themselves to avoid
+    side effects.
+
+    Args:
+        num_readers: The number of readers that will read from this channel. Readers
+            can be the same method of the same actor.
+    """
+
+    def __init__(
+        self,
+        num_readers,
+        _channel_id: Optional[str] = None,
+    ):
+        # Generate a unique ID for the channel. The writer and reader will use
+        # this ID to store and retrieve data from the _SerializationContext.
+        self._channel_id = _channel_id
+        self._num_readers = num_readers
+        if self._channel_id is None:
+            self._channel_id = str(uuid.uuid4())
+
+    def ensure_registered_as_writer(self) -> None:
+        pass
+
+    def ensure_registered_as_reader(self) -> None:
+        pass
+
+    def __reduce__(self):
+        return IntraProcessChannel, (
+            self._num_readers,
+            self._channel_id,
+        )
+
+    def __str__(self) -> str:
+        return f"IntraProcessChannel(channel_id={self._channel_id})"
+
+    def write(self, value: Any, timeout: Optional[float] = None):
+        self.ensure_registered_as_writer()
+        # No need to check timeout as the operation is non-blocking.
+
+        # Because both the reader and writer are in the same worker process,
+        # we can directly store the data in the context instead of storing
+        # it in the channel object. This removes the serialization overhead of `value`.
+        ctx = ChannelContext.get_current().serialization_context
+        ctx.set_data(self._channel_id, value, self._num_readers)
+
+    def read(self, timeout: Optional[float] = None, deserialize: bool = True) -> Any:
+        self.ensure_registered_as_reader()
+        assert deserialize, "Data passed from the actor to itself is never serialized"
+        # No need to check timeout as the operation is non-blocking.
+        ctx = ChannelContext.get_current().serialization_context
+        return ctx.get_data(self._channel_id)
+
+    def close(self) -> None:
+        ctx = ChannelContext.get_current().serialization_context
+        ctx.reset_data(self._channel_id)

.venv/lib/python3.11/site-packages/ray/experimental/channel/serialization_context.py

@@ -0,0 +1,174 @@
+import warnings
+from typing import TYPE_CHECKING, Any, Dict, List, Set, Tuple, Union
+
+if TYPE_CHECKING:
+    import numpy as np
+    import torch
+
+
+_TORCH_WARNING_FILTER_ACTIVATE = True
+
+
+class _SerializationContext:
+    def __init__(self):
+        # If true, then tensors found in the data to serialize are extracted
+        # and the caller should send them through an external transport.
+        self._use_external_transport: bool = False
+        # If _use_external_transport is True, then these are
+        # the tensors that should be sent or received
+        # out-of-band, through the external transport.
+        self._out_of_band_tensors: List["torch.Tensor"] = []
+        # During serialization, tensors sent out-of-band are replaced with
+        # integer placeholders. This tracks the set of placeholders seen.
+        self._deserialized_tensor_placeholders: Set[int] = set()
+
+        # Buffer for transferring data between tasks in the same worker process.
+        # The key is the channel ID, and the value is the data. We don't use a
+        # lock when reading/writing the buffer because a DAG node actor will only
+        # execute one task at a time in `do_exec_tasks`. It will not execute multiple
+        # Ray tasks on a single actor simultaneously.
+        self.intra_process_channel_buffers: Dict[str, Any] = {}
+        # The number of readers for each channel. When the number of readers
+        # reaches 0, remove the data from the buffer.
+        self.channel_id_to_num_readers: Dict[str, int] = {}
+
+    def set_data(self, channel_id: str, value: Any, num_readers: int) -> None:
+        assert num_readers > 0, "num_readers must be greater than 0."
+        assert (
+            channel_id not in self.intra_process_channel_buffers
+        ), f"Channel {channel_id} already exists in the buffer."
+        assert (
+            channel_id not in self.channel_id_to_num_readers
+        ), f"Channel {channel_id} already exists in the channel_id_to_num_readers."
+
+        self.intra_process_channel_buffers[channel_id] = value
+        self.channel_id_to_num_readers[channel_id] = num_readers
+
+    def has_data(self, channel_id: str) -> bool:
+        return channel_id in self.intra_process_channel_buffers
+
+    def get_data(self, channel_id: str) -> Any:
+        assert (
+            channel_id in self.intra_process_channel_buffers
+        ), f"Channel {channel_id} does not exist in the buffer."
+        assert (
+            channel_id in self.channel_id_to_num_readers
+        ), f"Channel {channel_id} does not exist in the channel_id_to_num_readers."
+
+        self.channel_id_to_num_readers[channel_id] -= 1
+        if self.channel_id_to_num_readers[channel_id] == 0:
+            # All readers have read the data, so we can remove it.
+            self.channel_id_to_num_readers.pop(channel_id)
+            return self.intra_process_channel_buffers.pop(channel_id)
+        return self.intra_process_channel_buffers[channel_id]
+
+    def reset_data(self, channel_id: str) -> None:
+        self.intra_process_channel_buffers.pop(channel_id, None)
+        self.channel_id_to_num_readers.pop(channel_id, None)
+
+    def set_use_external_transport(self, use_external_transport: bool) -> None:
+        self._use_external_transport = use_external_transport
+
+    def reset_out_of_band_tensors(
+        self, tensors: List["torch.Tensor"]
+    ) -> Tuple[List["torch.Tensor"], Set[int]]:
+        """
+        Return and reset the out-of-band tensors and all tensor placeholders
+        that were deserialized since the last call to reset.
+        """
+        prev_tensors = self._out_of_band_tensors
+        deserialized_tensor_placeholders = self._deserialized_tensor_placeholders
+        self._out_of_band_tensors = tensors
+        self._deserialized_tensor_placeholders = set()
+        return prev_tensors, deserialized_tensor_placeholders
+
+    def serialize_tensor(self, tensor: "torch.Tensor") -> Union[int, "np.ndarray"]:
+        from ray.experimental.channel import ChannelContext
+
+        ctx = ChannelContext.get_current()
+        if self._use_external_transport and tensor.device == ctx.torch_device:
+            # External transport is enabled and we found a tensor that matches
+            # our device.  Add the actual tensor to a buffer. The buffer of
+            # tensors should later be popped by the caller and sent via
+            # external transport.
+            self._out_of_band_tensors.append(tensor)
+            # Return a placeholder.
+            return len(self._out_of_band_tensors) - 1
+
+        return self.serialize_to_numpy(tensor)
+
+    def serialize_to_numpy(
+        self, tensor: "torch.Tensor"
+    ) -> Tuple["np.ndarray", "torch.dtype"]:
+        import torch
+
+        # Transfer through Ray's shared memory store for now.
+        # TODO(swang): This requires two copies, one to transfer from GPU to
+        # CPU and another from CPU to shared memory. Ideally we should elide
+        # the first copy and memcpy directly from GPU to the shared memory
+        # buffer.
+        if tensor.device.type == "cuda":
+            tensor = tensor.to("cpu")
+
+        # Numpy does not have an equivalent dtype for all torch dtypes, so
+        # instead of casting directly to numpy, we first use a view with a
+        # common dtype and then view as numpy array.
+        return (tensor.view(torch.uint8).numpy(), tensor.dtype)
+
+    def deserialize_tensor(self, val: Union["np.ndarray", int]):
+        # Found a placeholder for a tensor that was serialized via NCCL.
+        # Replace it with the corresponding deserialized tensor.
+        if isinstance(val, int):
+            placeholder = val
+            self._deserialized_tensor_placeholders.add(placeholder)
+            assert placeholder < len(self._out_of_band_tensors)
+            return self._out_of_band_tensors[placeholder]
+
+        return self.deserialize_from_numpy(val)
+
+    def deserialize_from_numpy(
+        self, np_array_dtype: Tuple["np.ndarray", "torch.dtype"]
+    ):
+        import torch
+
+        from ray.experimental.channel import ChannelContext
+
+        ctx = ChannelContext.get_current()
+
+        np_array, dtype = np_array_dtype
+
+        # TODO(swang): Support local P2P transfers if available.
+        # If there is a GPU assigned to this worker, move it there.
+        if ctx.torch_device is not None and ctx.torch_device.type == "cuda":
+
+            def convert_numpy_to_tensor(np_array, ctx):
+                # It does zero-copy convert np_array inside shared memroy to
+                # a tensor. Since we move data to GPU immediately, it is safe.
+                cpu_tensor = torch.from_numpy(np_array).view(dtype)
+                return cpu_tensor.to(device=ctx.torch_device)
+
+            global _TORCH_WARNING_FILTER_ACTIVATE
+            # filtering warning messages would be the bottleneck for
+            # deserializing torch tensors. Since the warning only prompts once,
+            # we would only deal with it for the first time.
+            if _TORCH_WARNING_FILTER_ACTIVATE:
+                with warnings.catch_warnings():
+                    # Since np_array.is_writable is False (it is set by Ray),
+                    # this raises a warning. Suppress it.
+                    warnings.filterwarnings(
+                        "ignore",
+                        category=UserWarning,
+                        message="The given NumPy array is not writable",
+                    )
+                    # gpu_tensor = convert_numpy_to_tensor(np_array, ctx)
+                    gpu_tensor = convert_numpy_to_tensor(np_array, ctx)
+                _TORCH_WARNING_FILTER_ACTIVATE = False
+            else:
+                gpu_tensor = convert_numpy_to_tensor(np_array, ctx)
+
+            return gpu_tensor
+
+        # TODO(swang): Use zero-copy from_numpy() if np_array.flags.writeable
+        # is True. This is safe to set when deserializing np_array if the
+        # upstream task has num_readers=1.
+        return torch.tensor(np_array, device=ctx.torch_device).view(dtype)

.venv/lib/python3.11/site-packages/ray/experimental/compiled_dag_ref.py

@@ -0,0 +1,227 @@
+import asyncio
+from typing import Any, List, Optional
+
+import ray
+from ray.exceptions import (
+    GetTimeoutError,
+    RayChannelError,
+    RayChannelTimeoutError,
+    RayTaskError,
+)
+from ray.util.annotations import PublicAPI
+
+
+def _process_return_vals(return_vals: List[Any], return_single_output: bool):
+    """
+    Process return values for return to the DAG caller. Any exceptions found in
+    return_vals will be raised. If return_single_output=True, it indicates that
+    the original DAG did not have a MultiOutputNode, so the DAG caller expects
+    a single return value instead of a list.
+    """
+    # Check for exceptions.
+    if isinstance(return_vals, Exception):
+        raise return_vals
+
+    for val in return_vals:
+        if isinstance(val, RayTaskError):
+            raise val.as_instanceof_cause()
+
+    if return_single_output:
+        assert len(return_vals) == 1
+        return return_vals[0]
+
+    return return_vals
+
+
+@PublicAPI(stability="alpha")
+class CompiledDAGRef:
+    """
+    A reference to a compiled DAG execution result.
+
+    This is a subclass of ObjectRef and resembles ObjectRef.  For example,
+    similar to ObjectRef, ray.get() can be called on it to retrieve the result.
+    However, there are several major differences:
+    1. ray.get() can only be called once per CompiledDAGRef.
+    2. ray.wait() is not supported.
+    3. CompiledDAGRef cannot be copied, deep copied, or pickled.
+    4. CompiledDAGRef cannot be passed as an argument to another task.
+    """
+
+    def __init__(
+        self,
+        dag: "ray.experimental.CompiledDAG",
+        execution_index: int,
+        channel_index: Optional[int] = None,
+    ):
+        """
+        Args:
+            dag: The compiled DAG that generated this CompiledDAGRef.
+            execution_index: The index of the execution for the DAG.
+                A DAG can be executed multiple times, and execution index
+                indicates which execution this CompiledDAGRef corresponds to.
+            actor_execution_loop_refs: The actor execution loop refs that
+                are used to execute the DAG. This can be used internally to
+                check the task execution errors in case of exceptions.
+            channel_index: The index of the DAG's output channel to fetch
+                the result from. A DAG can have multiple output channels, and
+                channel index indicates which channel this CompiledDAGRef
+                corresponds to. If channel index is not provided, this CompiledDAGRef
+                wraps the results from all output channels.
+
+        """
+        self._dag = dag
+        self._execution_index = execution_index
+        self._channel_index = channel_index
+        # Whether ray.get() was called on this CompiledDAGRef.
+        self._ray_get_called = False
+        self._dag_output_channels = dag.dag_output_channels
+
+    def __str__(self):
+        return (
+            f"CompiledDAGRef({self._dag.get_id()}, "
+            f"execution_index={self._execution_index}, "
+            f"channel_index={self._channel_index})"
+        )
+
+    def __copy__(self):
+        raise ValueError("CompiledDAGRef cannot be copied.")
+
+    def __deepcopy__(self, memo):
+        raise ValueError("CompiledDAGRef cannot be deep copied.")
+
+    def __reduce__(self):
+        raise ValueError("CompiledDAGRef cannot be pickled.")
+
+    def __del__(self):
+        # If the dag is already teardown, it should do nothing.
+        if self._dag.is_teardown:
+            return
+
+        # If we've already cached the result in the buffer, get it to remove it
+        # from the buffer. Else add this CompiledDAGRef's execution and channel indices
+        # to the dag's _destructed_ref_idxs, and try to release any buffers we
+        # can based on the dag's current max_finished_execution_index.
+        if not self._ray_get_called:
+            if self._execution_index in self._dag._result_buffer:
+                self._dag._get_execution_results(
+                    self._execution_index, self._channel_index
+                )
+            else:
+                self._dag._destructed_ref_idxs[self._execution_index].add(
+                    self._channel_index
+                )
+                self._dag._try_release_buffers()
+
+    def get(self, timeout: Optional[float] = None):
+        if self._ray_get_called:
+            raise ValueError(
+                "ray.get() can only be called once "
+                "on a CompiledDAGRef, and it was already called."
+            )
+
+        self._ray_get_called = True
+        try:
+            self._dag._execute_until(
+                self._execution_index, self._channel_index, timeout
+            )
+            return_vals = self._dag._get_execution_results(
+                self._execution_index, self._channel_index
+            )
+        except RayChannelTimeoutError:
+            raise
+        except RayChannelError as channel_error:
+            # If we get a channel error, we'd like to call ray.get() on
+            # the actor execution loop refs to check if this is a result
+            # of task execution error which could not be passed down
+            # (e.g., when a pure NCCL channel is used, it is only
+            # able to send tensors, but not the wrapped exceptions).
+            # In this case, we'd like to raise the task execution error
+            # (which is the actual cause of the channel error) instead
+            # of the channel error itself.
+            # TODO(rui): determine which error to raise if multiple
+            # actor task refs have errors.
+            actor_execution_loop_refs = list(self._dag.worker_task_refs.values())
+            try:
+                ray.get(actor_execution_loop_refs, timeout=10)
+            except GetTimeoutError as timeout_error:
+                raise Exception(
+                    "Timed out when getting the actor execution loop exception. "
+                    "This should not happen, please file a GitHub issue."
+                ) from timeout_error
+            except Exception as execution_error:
+                # Use 'from None' to suppress the context of the original
+                # channel error, which is not useful to the user.
+                raise execution_error from None
+            else:
+                raise channel_error
+        except Exception:
+            raise
+        return _process_return_vals(return_vals, True)
+
+
+@PublicAPI(stability="alpha")
+class CompiledDAGFuture:
+    """
+    A reference to a compiled DAG execution result, when executed with asyncio.
+    This differs from CompiledDAGRef in that `await` must be called on the
+    future to get the result, instead of `ray.get()`.
+
+    This resembles async usage of ObjectRefs. For example, similar to
+    ObjectRef, `await` can be called directly on the CompiledDAGFuture to
+    retrieve the result.  However, there are several major differences:
+    1. `await` can only be called once per CompiledDAGFuture.
+    2. ray.wait() is not supported.
+    3. CompiledDAGFuture cannot be copied, deep copied, or pickled.
+    4. CompiledDAGFuture cannot be passed as an argument to another task.
+    """
+
+    def __init__(
+        self,
+        dag: "ray.experimental.CompiledDAG",
+        execution_index: int,
+        fut: "asyncio.Future",
+        channel_index: Optional[int] = None,
+    ):
+        self._dag = dag
+        self._execution_index = execution_index
+        self._fut = fut
+        self._channel_index = channel_index
+
+    def __str__(self):
+        return (
+            f"CompiledDAGFuture({self._dag.get_id()}, "
+            f"execution_index={self._execution_index}, "
+            f"channel_index={self._channel_index})"
+        )
+
+    def __copy__(self):
+        raise ValueError("CompiledDAGFuture cannot be copied.")
+
+    def __deepcopy__(self, memo):
+        raise ValueError("CompiledDAGFuture cannot be deep copied.")
+
+    def __reduce__(self):
+        raise ValueError("CompiledDAGFuture cannot be pickled.")
+
+    def __await__(self):
+        if self._fut is None:
+            raise ValueError(
+                "CompiledDAGFuture can only be awaited upon once, and it has "
+                "already been awaited upon."
+            )
+
+        # NOTE(swang): If the object is zero-copy deserialized, then it will
+        # stay in scope as long as this future is in scope. Therefore, we
+        # delete self._fut here before we return the result to the user.
+        fut = self._fut
+        self._fut = None
+
+        if not self._dag._has_execution_results(self._execution_index):
+            result = yield from fut.__await__()
+            self._dag._max_finished_execution_index += 1
+            self._dag._cache_execution_results(self._execution_index, result)
+
+        return_vals = self._dag._get_execution_results(
+            self._execution_index, self._channel_index
+        )
+        return _process_return_vals(return_vals, True)

.venv/lib/python3.11/site-packages/ray/experimental/gradio_utils.py

@@ -0,0 +1,12 @@
+def type_to_string(_type: type) -> str:
+    """Gets the string representation of a type.
+
+    THe original type can be derived from the returned string representation through
+    pydoc.locate().
+    """
+    if _type.__module__ == "typing":
+        return f"{_type.__module__}.{_type._name}"
+    elif _type.__module__ == "builtins":
+        return _type.__name__
+    else:
+        return f"{_type.__module__}.{_type.__name__}"

.venv/lib/python3.11/site-packages/ray/experimental/internal_kv.py

@@ -0,0 +1,121 @@
+from typing import List, Optional, Union
+
+from ray._private.client_mode_hook import client_mode_hook
+from ray._raylet import GcsClient
+
+_initialized = False
+global_gcs_client = None
+
+
+def _internal_kv_reset():
+    global global_gcs_client, _initialized
+    global_gcs_client = None
+    _initialized = False
+
+
+def internal_kv_get_gcs_client():
+    return global_gcs_client
+
+
+def _initialize_internal_kv(gcs_client: GcsClient):
+    """Initialize the internal KV for use in other function calls."""
+    global global_gcs_client, _initialized
+    assert gcs_client is not None
+    global_gcs_client = gcs_client
+    _initialized = True
+
+
+@client_mode_hook
+def _internal_kv_initialized():
+    return global_gcs_client is not None
+
+
+@client_mode_hook
+def _internal_kv_get(
+    key: Union[str, bytes], *, namespace: Optional[Union[str, bytes]] = None
+) -> bytes:
+    """Fetch the value of a binary key."""
+
+    if isinstance(key, str):
+        key = key.encode()
+    if isinstance(namespace, str):
+        namespace = namespace.encode()
+    assert isinstance(key, bytes)
+    return global_gcs_client.internal_kv_get(key, namespace)
+
+
+@client_mode_hook
+def _internal_kv_exists(
+    key: Union[str, bytes], *, namespace: Optional[Union[str, bytes]] = None
+) -> bool:
+    """Check key exists or not."""
+
+    if isinstance(key, str):
+        key = key.encode()
+    if isinstance(namespace, str):
+        namespace = namespace.encode()
+    assert isinstance(key, bytes)
+    return global_gcs_client.internal_kv_exists(key, namespace)
+
+
+@client_mode_hook
+def _pin_runtime_env_uri(uri: str, *, expiration_s: int) -> None:
+    """Pin a runtime_env URI for expiration_s."""
+    return global_gcs_client.pin_runtime_env_uri(uri, expiration_s)
+
+
+@client_mode_hook
+def _internal_kv_put(
+    key: Union[str, bytes],
+    value: Union[str, bytes],
+    overwrite: bool = True,
+    *,
+    namespace: Optional[Union[str, bytes]] = None
+) -> bool:
+    """Globally associates a value with a given binary key.
+
+    This only has an effect if the key does not already have a value.
+
+    Returns:
+        already_exists: whether the value already exists.
+    """
+
+    if isinstance(key, str):
+        key = key.encode()
+    if isinstance(value, str):
+        value = value.encode()
+    if isinstance(namespace, str):
+        namespace = namespace.encode()
+    assert (
+        isinstance(key, bytes)
+        and isinstance(value, bytes)
+        and isinstance(overwrite, bool)
+    )
+    return global_gcs_client.internal_kv_put(key, value, overwrite, namespace) == 0
+
+
+@client_mode_hook
+def _internal_kv_del(
+    key: Union[str, bytes],
+    *,
+    del_by_prefix: bool = False,
+    namespace: Optional[Union[str, bytes]] = None
+) -> int:
+    if isinstance(key, str):
+        key = key.encode()
+    if isinstance(namespace, str):
+        namespace = namespace.encode()
+    assert isinstance(key, bytes)
+    return global_gcs_client.internal_kv_del(key, del_by_prefix, namespace)
+
+
+@client_mode_hook
+def _internal_kv_list(
+    prefix: Union[str, bytes], *, namespace: Optional[Union[str, bytes]] = None
+) -> List[bytes]:
+    """List all keys in the internal KV store that start with the prefix."""
+    if isinstance(prefix, str):
+        prefix = prefix.encode()
+    if isinstance(namespace, str):
+        namespace = namespace.encode()
+    return global_gcs_client.internal_kv_keys(prefix, namespace)