Fix struct generation in execution file (#434)

flyteorg · Oct 19, 2023 · cdfa222 · cdfa222
1 parent 043a611
commit cdfa222
Show file tree

Hide file tree

Showing 2 changed files with 67 additions and 10 deletions.
diff --git a/cmd/get/execution_util.go b/cmd/get/execution_util.go
@@ -6,6 +6,8 @@ import (
 	"io/ioutil"
 	"os"
 
+	structpb "github.com/golang/protobuf/ptypes/struct"
+
 	"gopkg.in/yaml.v3"
 
 	cmdUtil "github.com/flyteorg/flytectl/pkg/commandutils"
@@ -152,7 +154,14 @@ func ParamMapForWorkflow(lp *admin.LaunchPlan) (map[string]yaml.Node, error) {
 
 func getCommentedYamlNode(input interface{}, comment string) (yaml.Node, error) {
 	var node yaml.Node
-	err := node.Encode(input)
+	var err error
+
+	if s, ok := input.(*structpb.Struct); ok {
+		err = node.Encode(s.AsMap())
+	} else {
+		err = node.Encode(input)
+	}
+
 	node.LineComment = comment
 	return node, err
 }