From 07b1058a10c0a4dab901cd799eaa5de2aa7dc7cf Mon Sep 17 00:00:00 2001
From: Ilija Vukotic <ivukotic@uchicago.edu>
Date: Wed, 15 Nov 2023 18:59:42 -0600
Subject: [PATCH] Add Kubernetes pod templates for Dask scheduler and worker

---
 dask_config.yaml | 92 +++++++++++++++++++++++-------------------------
 1 file changed, 45 insertions(+), 47 deletions(-)

diff --git a/dask_config.yaml b/dask_config.yaml
index 2765b73..37c128e 100644
--- a/dask_config.yaml
+++ b/dask_config.yaml
@@ -93,52 +93,50 @@ kubernetes:
   worker-template-path: null
 
   scheduler-template:
-    {}
-    # kind: Pod
-    # metadata:
-    #   labels:
-    #     foo: bar
-    #     baz: quux
-    # spec:
-    #   restartPolicy: Never
-    #   containers:
-    #   - -scheduler
-    #     image: ghcr.io/dask/dask:latest
-    #     args:
-    #       - dask-scheduler
-    #     resources:
-    #       limits:
-    #         cpu: "1.75"
-    #         memory: 6G
-    #       requests:
-  #         cpu: "1.75"
-  #         memory: 6G
+    kind: Pod
+    metadata:
+      labels:
+        foo: bar
+        baz: quux
+    spec:
+      restartPolicy: Never
+      containers:
+        - name: dask-scheduler
+          image: ghcr.io/dask/dask:latest
+          args:
+            - dask-scheduler
+          resources:
+            limits:
+              cpu: "1.75"
+              memory: 6G
+            requests:
+              cpu: "1.75"
+              memory: 6G
 
   worker-template:
-    {}
-    # kind: Pod
-    # metadata:
-    #   labels:
-    #     foo: bar
-    #     baz: quux
-    # spec:
-    #   restartPolicy: Never
-    #   containers:
-    #   - name: dask-worker
-    #     image: ghcr.io/dask/dask:latest
-    #     args:
-    #       - dask-worker
-    #       - --nthreads
-    #       - '2'
-    #       - --no-bokeh
-    #       - --memory-limit
-    #       - 6GB
-    #       - --death-timeout
-    #       - '60'
-    #     resources:
-    #       limits:
-    #         cpu: "1.75"
-    #         memory: 6G
-    #       requests:
-    #         cpu: "1.75"
-    #         memory: 6G
+    kind: Pod
+    metadata:
+      labels:
+        foo: bar
+        baz: quux
+    spec:
+      restartPolicy: Never
+      containers:
+        - name: dask-worker
+          image: ghcr.io/dask/dask:latest
+          args:
+            - dask-worker
+            - --nthreads
+            - "2"
+            - --no-bokeh
+            - --memory-limit
+            - 6GB
+            - --death-timeout
+            - "60"
+          resources:
+            limits:
+              cpu: "1.75"
+              memory: 6G
+            requests:
+              cpu: "1.75"
+              memory: 6G