adding IsArray flag to NodeExecutionMetadata

Signed-off-by: Daniel Rammer <[email protected]>
flyteorg · Oct 4, 2023 · 4a80465 · 4a80465
1 parent 54902c7
commit 4a80465
Show file tree

Hide file tree

Showing 44 changed files with 1,016 additions and 541 deletions.
diff --git a/flyteadmin/pkg/repositories/transformers/node_execution.go b/flyteadmin/pkg/repositories/transformers/node_execution.go
@@ -131,6 +131,7 @@ func CreateNodeExecutionModel(ctx context.Context, input ToNodeExecutionModelInp
 		SpecNodeId:   input.Request.Event.SpecNodeId,
 		IsParentNode: input.Request.Event.IsParent,
 		IsDynamic:    input.Request.Event.IsDynamic,
+		IsArray:      input.Request.Event.IsArray,
 	}
 	err := handleNodeExecutionInputs(ctx, nodeExecution, input.Request, input.StorageClient)
 	if err != nil {
@@ -304,6 +305,9 @@ func UpdateNodeExecutionModel(
 		if request.Event.IsDynamic {
 			nodeExecutionMetadata.IsDynamic = true
 		}
+		if request.Event.IsArray {
+			nodeExecutionMetadata.IsArray = true
+		}
 		nodeExecMetadataBytes, err := proto.Marshal(&nodeExecutionMetadata)
 		if err != nil {
 			return errors.NewFlyteAdminErrorf(codes.Internal,

diff --git a/flyteidl/gen/pb-cpp/flyteidl/admin/node_execution.pb.cc b/flyteidl/gen/pb-cpp/flyteidl/admin/node_execution.pb.cc
diff --git a/flyteidl/gen/pb-cpp/flyteidl/admin/node_execution.pb.h b/flyteidl/gen/pb-cpp/flyteidl/admin/node_execution.pb.h