Skip to content

Commit

Permalink
Ray task proposal (#308)
Browse files Browse the repository at this point in the history
* Ray plugin

Signed-off-by: Kevin Su <[email protected]>

* Ray plugin

Signed-off-by: Kevin Su <[email protected]>

* Update cluster spec

Signed-off-by: Kevin Su <[email protected]>

* Update cluster spec

Signed-off-by: Kevin Su <[email protected]>

* Update proto

Signed-off-by: Kevin Su <[email protected]>

* Update proto

Signed-off-by: Kevin Su <[email protected]>

* Update proto

Signed-off-by: Kevin Su <[email protected]>

* Update proto

Signed-off-by: Kevin Su <[email protected]>

* Update proto

Signed-off-by: Kevin Su <[email protected]>

* Fix lint error

Signed-off-by: Kevin Su <[email protected]>

* Remove shutdown_after_job_finishes

Signed-off-by: Kevin Su <[email protected]>

* More comments

Signed-off-by: Kevin Su <[email protected]>
  • Loading branch information
pingsutw authored Aug 6, 2022
1 parent 3905f8a commit a457ef3
Show file tree
Hide file tree
Showing 14 changed files with 8,525 additions and 3 deletions.
1 change: 0 additions & 1 deletion flyteidl/boilerplate/update.cfg
Original file line number Diff line number Diff line change
@@ -1,5 +1,4 @@
flyte/golang_test_targets
flyte/golang_support_tools
flyte/pull_request_template
flyte/welcome_bot
flyte/code_of_conduct
2 changes: 1 addition & 1 deletion flyteidl/clients/go/admin/pkce/auth_flow_orchestrator.go
Original file line number Diff line number Diff line change
Expand Up @@ -112,7 +112,7 @@ func (f TokenOrchestrator) FetchTokenFromAuthFlow(ctx context.Context) (*oauth2.
pkceCodeChallenge + "&code_challenge_method=S256"

serveMux := http.NewServeMux()
server := &http.Server{Addr: redirectURL.Host, Handler: serveMux}
server := &http.Server{Addr: redirectURL.Host, Handler: serveMux, ReadHeaderTimeout: 0}
// Register the call back handler
serveMux.HandleFunc(redirectURL.Path, getAuthServerCallbackHandler(f.clientConfig, pkceCodeVerifier,
tokenChannel, errorChannel, stateString)) // the oauth2 callback endpoint
Expand Down
24 changes: 24 additions & 0 deletions flyteidl/gen/pb-cpp/flyteidl/plugins/ray.grpc.pb.cc

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

47 changes: 47 additions & 0 deletions flyteidl/gen/pb-cpp/flyteidl/plugins/ray.grpc.pb.h

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

2,062 changes: 2,062 additions & 0 deletions flyteidl/gen/pb-cpp/flyteidl/plugins/ray.pb.cc

Large diffs are not rendered by default.

Loading

0 comments on commit a457ef3

Please sign in to comment.