-
Notifications
You must be signed in to change notification settings - Fork 53
Conversation
This is great stuff @igorvalko, once you change the flyteidl, you have a +1 from me |
workers := pytorchTaskExtraArgs.GetWorkers() | ||
|
||
jobSpec := ptOp.PyTorchJobSpec{ | ||
TTLSecondsAfterFinished: nil, //60, // cleanup in 60 seconds after the job completion |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
ya you dont need to cleanup, we will automatically
"github.com/lyft/flyteplugins/go/tasks/logs" | ||
logUtils "github.com/lyft/flyteidl/clients/go/coreutils/logs" | ||
|
||
//commonOp "github.com/kubeflow/common/pkg/apis/common/v1" // switch to real 'common' once https://github.com/kubeflow/pytorch-operator/issues/263 resolved |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
interesting call out. I think after this for all kubeflow operators, we can just have a flyte-kubeflow-commons
where all status checking can be done inline
Codecov Report
@@ Coverage Diff @@
## master #85 +/- ##
==========================================
+ Coverage 54.97% 55.92% +0.94%
==========================================
Files 91 93 +2
Lines 4660 4787 +127
==========================================
+ Hits 2562 2677 +115
- Misses 1821 1827 +6
- Partials 277 283 +6
Continue to review full report at Codecov.
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
TL;DR
Pytorch plugin for Flyte that leverages https://github.com/kubeflow/pytorch-operator to run distributed pytorch jobs on k8s.
Type
Are all requirements met?
Complete description
Pytorch plugin for Flyte that leverages https://github.com/kubeflow/pytorch-operator to run distributed pytorch jobs on k8s.
Related PRs:
flyteorg/flyteidl#61
flyteorg/flytekit#112
Tracking Issue
NA
Follow-up issue
NA