Skip to content

Commit a98c7ab

Browse files
committed
feat(sdk) Add Input Parameter support for node affinity
Signed-off-by: VaniHaripriya <[email protected]>
1 parent ada935a commit a98c7ab

File tree

3 files changed

+328
-85
lines changed

3 files changed

+328
-85
lines changed

kubernetes_platform/go/kubernetesplatform/kubernetes_executor_config.pb.go

Lines changed: 91 additions & 71 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

kubernetes_platform/proto/kubernetes_executor_config.proto

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -241,7 +241,14 @@ message NodeAffinityTerm {
241241
repeated SelectorRequirement match_expressions = 1;
242242
repeated SelectorRequirement match_fields = 2;
243243
//Setting the weight makes it use PreferredDuringSchedulingIgnoredDuringExecution rules instead of RequiredDuringSchedulingIgnoredDuringExecution rules
244-
optional int32 weight = 3;
244+
optional int32 weight = 3;
245+
// Provide a JSON struct of node affinity. Takes precedence over PreferredDuringSchedulingIgnoredDuringExecution rules/RequiredDuringSchedulingIgnoredDuringExecution rules.
246+
// The JSON must follow Kubernetes
247+
// NodeAffinity structure:
248+
// https://kubernetes.io/docs/reference/generated/kubernetes-api/v1.27/#nodeaffinity-v1-core
249+
// Example:
250+
// { "requiredDuringSchedulingIgnoredDuringExecution": { "nodeSelectorTerms": [ ... ] } }
251+
ml_pipelines.TaskInputsSpec.InputParameterSpec node_affinity_json = 4;
245252
}
246253

247254
message PodAffinityTerm {

0 commit comments

Comments
 (0)