feat: EFS compatibility Storage Class for AWS (#55)

wandb · Dec 18, 2023 · bae5869 · bae5869
1 parent ac17a5f
commit bae5869
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 1 deletion.
diff --git a/charts/operator-wandb/Chart.yaml b/charts/operator-wandb/Chart.yaml
@@ -2,7 +2,7 @@ apiVersion: v2
 name: operator-wandb
 description: A Helm chart for deploying W&B to Kubernetes
 type: application
-version: 0.10.30
+version: 0.10.31
 appVersion: 1.0.0
 icon: https://wandb.ai/logo.svg
 

diff --git a/charts/operator-wandb/charts/weave/templates/pvc.yaml b/charts/operator-wandb/charts/weave/templates/pvc.yaml
@@ -52,5 +52,19 @@ mountOptions:
   - "strictatime"
 reclaimPolicy: Delete
 volumeBindingMode: WaitForFirstConsumer
+{{- else if eq .Values.persistence.provider "efs" }}
+provisioner: efs.csi.aws.com
+reclaimPolicy: Delete
+allowVolumeExpansion: true
+parameters:
+  provisioningMode: efs-ap
+  fileSystemId: {{ .Values.persistence.efs.fileSystemId | quote }}
+  directoryPerms: "700"
+  gidRangeStart: "1000" # optional
+  gidRangeEnd: "2000" # optional
+  basePath: "/dynamic_provisioning" # optional
+  ensureUniqueDirectory: "true" # optional
+  reuseAccessPoint: "false" # optional
+volumeBindingMode: WaitForFirstConsumer
 {{- end }}
 {{- end }}
diff --git a/charts/operator-wandb/charts/weave/values.yaml b/charts/operator-wandb/charts/weave/values.yaml
@@ -22,6 +22,7 @@ clusterRole: {}
 
 service:
   type: ClusterIP
+  annotations: {}
 
 pod:
   securityContext:
@@ -49,3 +50,5 @@ persistence:
   accessMode: ReadWriteMany
   # storageClassName: ""
   size: 20Gi
+  efs:
+    fileSystemId: ""
diff --git a/charts/operator-wandb/values.yaml b/charts/operator-wandb/values.yaml
@@ -105,6 +105,9 @@ weave:
   image:
     repository: wandb/local
     tag: latest
+  persistence:
+    efs:
+      fileSystemId: ""
 
 console:
   install: true