Add ephemeral storage to node resource requirements

Signed-off-by: Katrina Rogan <katroganGH@gmail.com>
flyteorg · Aug 10, 2021 · 7bd2037 · 7bd2037
1 parent 5fe91f3
commit 7bd2037
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/pkg/utils/k8s.go b/pkg/utils/k8s.go
@@ -57,6 +57,10 @@ func ToK8sResourceList(resources []*core.Resources_ResourceEntry) (v1.ResourceLi
 			if !v.IsZero() {
 				k8sResources[ResourceNvidiaGPU] = v
 			}
+		case core.Resources_EPHEMERAL_STORAGE:
+			if !v.IsZero() {
+				k8sResources[v1.ResourceEphemeralStorage] = v
+			}
 		}
 	}
 	return k8sResources, nil

diff --git a/pkg/utils/k8s_test.go b/pkg/utils/k8s_test.go
@@ -37,6 +37,7 @@ func TestToK8sResourceList(t *testing.T) {
 			{Name: core.Resources_GPU, Value: "1"},
 			{Name: core.Resources_MEMORY, Value: "1024Mi"},
 			{Name: core.Resources_STORAGE, Value: "1024Mi"},
+			{Name: core.Resources_EPHEMERAL_STORAGE, Value: "1024Mi"},
 		})
 
 		assert.NoError(t, err)
@@ -46,6 +47,7 @@ func TestToK8sResourceList(t *testing.T) {
 		assert.Equal(t, resource.MustParse("1"), r[ResourceNvidiaGPU])
 		assert.Equal(t, resource.MustParse("1024Mi"), r[v1.ResourceMemory])
 		assert.Equal(t, resource.MustParse("1024Mi"), r[v1.ResourceStorage])
+		assert.Equal(t, resource.MustParse("1024Mi"), r[v1.ResourceEphemeralStorage])
 	}
 	{
 		r, err := ToK8sResourceList([]*core.Resources_ResourceEntry{})