Open pabloem opened 1 year ago
@valiantljk this is something that you could consider taking a stab with? : )
you'd have to add some smart code so that ongoing ray_execute_bundle
tasks can report progress - a bit of Beam code to assemble the ProcessBundleProgressRequest
protos, and some Ray code to create actors/threads to report results
lmk if that helps
For how to do the ProcessBundleProgressRequest
s, you can check the original FnApiRunner
code, I think there's a basic implementation there.
The Beam API implements several RPCs for processing of data.
The main RPC is the
ProcessBundle
RPC, which we've implemented.During the execution of a
ProcessBundle
rpc, the Beam worker consumes data to process, and outputs data. While this request is being executed, the runner can request progress reports from the worker. These progress reports are requested via theProcessBundleProgress
rpc.This work item consists on:
ProcessBundleProgress
rpcs for the SDK workerSome ideas:
This is just one idea, but we can do things in different ways.