update array node model

Signed-off-by: Paul Dittamo <[email protected]>

flytekit/models/core/workflow.py

@@ -395,6 +395,8 @@ def __init__(
         execution_mode=None,
         is_original_sub_node_interface=False,
         data_mode=None,
+        union_array_node_spec=None,
+        fixed_inputs=None,
     ) -> None:
         """
         TODO: docstring
@@ -407,6 +409,8 @@ def __init__(
         self._execution_mode = execution_mode
         self._is_original_sub_node_interface = is_original_sub_node_interface
         self._data_mode = data_mode
+        self._union_array_node_spec = union_array_node_spec
+        self._fixed_inputs = fixed_inputs
 
     @property
     def node(self) -> "Node":
@@ -421,6 +425,8 @@ def to_flyte_idl(self) -> _core_workflow.ArrayNode:
             execution_mode=self._execution_mode,
             is_original_sub_node_interface=BoolValue(value=self._is_original_sub_node_interface),
             data_mode=self._data_mode,
+            union_array_node_spec=self._union_array_node_spec,
+            fixed_inputs=self._fixed_inputs,
         )
 
     @classmethod

flytekit/tools/translator.py

@@ -621,6 +621,8 @@ def get_serializable_array_node(
         execution_mode=array_node.execution_mode,
         is_original_sub_node_interface=array_node.is_original_sub_node_interface,
         data_mode=array_node.data_mode,
+        union_array_node_spec=getattr(array_node, "union_array_node_spec", None),
+        fixed_inputs=array_node.fixed_inputs,
     )