Add priorityClassName value to operator deployment (#845)

Co-authored-by: Petr Šebek <petr.sebek@second-foundation.eu> Co-authored-by: Jacob Tomlinson <jtomlinson@nvidia.com>
dask · Jan 5, 2024 · 90c7d9b · 90c7d9b
1 parent eeb3c40
commit 90c7d9b
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/dask_kubernetes/operator/deployment/helm/dask-kubernetes-operator/README.md b/dask_kubernetes/operator/deployment/helm/dask-kubernetes-operator/README.md
@@ -35,6 +35,7 @@ The following table lists the configurable parameters of the Dask-kubernetes-ope
 | `nodeSelector` | Node selector | `{}` |
 | `tolerations` | Tolerations | `[]` |
 | `affinity` | Affinity | `{}` |
+| `priorityClassName` | Priority class | `null` |
 | `kopfArgs` | Command line flags to pass to kopf on start up | `["--all-namespaces"]` |
 | `metrics.scheduler.enabled` | Enable scheduler metrics. Pip package [prometheus-client](https://pypi.org/project/prometheus-client/) should be present on scheduler. | `false` |
 | `metrics.scheduler.serviceMonitor.enabled` | Enable scheduler servicemonitor. | `false` |

diff --git a/dask_kubernetes/operator/deployment/helm/dask-kubernetes-operator/templates/deployment.yaml b/dask_kubernetes/operator/deployment/helm/dask-kubernetes-operator/templates/deployment.yaml
@@ -68,3 +68,6 @@ spec:
       tolerations:
         {{- toYaml . | nindent 8 }}
       {{- end }}
+      {{- with .Values.priorityClassName }}
+      priorityClassName: {{ . | quote  }}
+      {{- end }}
diff --git a/dask_kubernetes/operator/deployment/helm/dask-kubernetes-operator/values.yaml b/dask_kubernetes/operator/deployment/helm/dask-kubernetes-operator/values.yaml
@@ -52,6 +52,8 @@ tolerations: []  # Tolerations
 
 affinity: {}  # Affinity
 
+priorityClassName: null  # Priority class
+
 kopfArgs: # Command line flags to pass to kopf on start up
   - --all-namespaces