From 3b690d206155fa059f31e0d963ba9b120a25104a Mon Sep 17 00:00:00 2001
From: Yee Hing Tong <wild-endeavor@users.noreply.github.com>
Date: Sat, 24 Aug 2024 10:24:13 -0700
Subject: [PATCH] [flytectl] DataConfig missing from TaskSpec (#5692)

Signed-off-by: Yee Hing Tong <wild-endeavor@users.noreply.github.com>
Signed-off-by: Bugra Gedik <bgedik@gmail.com>
---
 flytectl/cmd/register/register_util.go | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/flytectl/cmd/register/register_util.go b/flytectl/cmd/register/register_util.go
index 0bf5a23d49..b7b419e611 100644
--- a/flytectl/cmd/register/register_util.go
+++ b/flytectl/cmd/register/register_util.go
@@ -339,8 +339,9 @@ func hydrateTaskSpec(task *admin.TaskSpec, sourceUploadedLocation storage.DataRe
 		}
 		task.Template.Target = &core.TaskTemplate_K8SPod{
 			K8SPod: &core.K8SPod{
-				Metadata: task.Template.GetK8SPod().Metadata,
-				PodSpec:  podSpecStruct,
+				Metadata:   task.Template.GetK8SPod().Metadata,
+				PodSpec:    podSpecStruct,
+				DataConfig: task.Template.GetK8SPod().DataConfig,
 			},
 		}
 	}