cuddly-jelly-27016
03/26/2025, 1:42 PMpod_template
to specify things like node selectors and tolerations. However, this is specific to certain workflows that share the core training tasks with other workflows. This means we have to duplicate the task function for each type of override we require. If we can support pod_template
in with_overrides
, this would reduce a lot of toil since we can supply pod templates in a central location and override downstream tasks, similar to how we can do so for resources.
### Goal: What should the final outcome look like, ideally?
The with_overrides
function here should be able to take in a pod_template
argument, similar to how it can take in requests
and limits
.
### Describe alternatives you've considered
We've duplicated each training function for every variation of pod template we have. This isn't very scalable though, so it restricts users from using many different specifications for different workflows.
### Propose: Link/Inline OR Additional context
No response
### Are you sure this issue hasn't been raised already?
• Yes
### Have you read the Code of Conduct?
• Yes
flyteorg/flytecuddly-jelly-27016
03/26/2025, 1:42 PM