Skip to content

Commit

Permalink
Add Ray Autoscaler to the Flyte-Ray plugin (#4363)
Browse files Browse the repository at this point in the history
* add enable_in_tree_autoscaling

Signed-off-by: Yicheng-Lu-llll <[email protected]>

* remove print

Signed-off-by: Yicheng-Lu-llll <[email protected]>

* allow MinReplicas to be zero

Signed-off-by: Yicheng-Lu-llll <[email protected]>

* fix ci test

Signed-off-by: Yicheng-Lu-llll <[email protected]>

* fix ci test

Signed-off-by: Yicheng-Lu-llll <[email protected]>

* add ttl

Signed-off-by: Yicheng-Lu-llll <[email protected]>

* nit

Signed-off-by: Yicheng-Lu-llll <[email protected]>

* change JobDeploymentStatusFailedToGetJobStatus -> PhaseInfoRunning

Signed-off-by: Yicheng-Lu-llll <[email protected]>

* rebasing

Signed-off-by: Yicheng-Lu-llll <[email protected]>

* rebasing

Signed-off-by: Yicheng-Lu-llll <[email protected]>

* enforce minreplica < replica < maxreplica

Signed-off-by: Yicheng-Lu-llll <[email protected]>

* add test for ShutdownAfterJobFinishes and TTLSecondsAfterFinished

Signed-off-by: Yicheng-Lu-llll <[email protected]>

* fix ray test error

Signed-off-by: Yicheng-Lu-llll <[email protected]>

---------

Signed-off-by: Yicheng-Lu-llll <[email protected]>
Co-authored-by: Kevin Su <[email protected]>
  • Loading branch information
Yicheng-Lu-llll and pingsutw authored Dec 26, 2023
1 parent 853ebe4 commit 0e35cab
Show file tree
Hide file tree
Showing 11 changed files with 668 additions and 132 deletions.
191 changes: 165 additions & 26 deletions flyteidl/gen/pb-cpp/flyteidl/plugins/ray.pb.cc

Large diffs are not rendered by default.

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

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

103 changes: 68 additions & 35 deletions flyteidl/gen/pb-go/flyteidl/plugins/ray.pb.go

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

6 changes: 6 additions & 0 deletions flyteidl/gen/pb-go/flyteidl/plugins/ray.pb.validate.go

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

Loading

0 comments on commit 0e35cab

Please sign in to comment.