Ray task proposal (#308)

* 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]>
flyteorg · Aug 6, 2022 · a457ef3 · a457ef3
1 parent 3905f8a
commit a457ef3
Show file tree

Hide file tree

Showing 14 changed files with 8,525 additions and 3 deletions.
diff --git a/flyteidl/boilerplate/update.cfg b/flyteidl/boilerplate/update.cfg
@@ -1,5 +1,4 @@
 flyte/golang_test_targets
 flyte/golang_support_tools
 flyte/pull_request_template
-flyte/welcome_bot
 flyte/code_of_conduct
diff --git a/flyteidl/clients/go/admin/pkce/auth_flow_orchestrator.go b/flyteidl/clients/go/admin/pkce/auth_flow_orchestrator.go
@@ -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

diff --git a/flyteidl/gen/pb-cpp/flyteidl/plugins/ray.grpc.pb.cc b/flyteidl/gen/pb-cpp/flyteidl/plugins/ray.grpc.pb.cc
diff --git a/flyteidl/gen/pb-cpp/flyteidl/plugins/ray.grpc.pb.h b/flyteidl/gen/pb-cpp/flyteidl/plugins/ray.grpc.pb.h
diff --git a/flyteidl/gen/pb-cpp/flyteidl/plugins/ray.pb.cc b/flyteidl/gen/pb-cpp/flyteidl/plugins/ray.pb.cc