add ArrayNodes to types

Signed-off-by: Jan Fiedler <jan@union.ai>
flyteorg · Apr 9, 2024 · f970b4e · f970b4e
1 parent 34edde1
commit f970b4e
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/packages/oss-console/src/models/Node/types.ts b/packages/oss-console/src/models/Node/types.ts
@@ -6,6 +6,15 @@ export type WorkflowNode = Core.IWorkflowNode;
 /** A graph node indicating a branching decision. */
 export type BranchNode = Core.IBranchNode;
 
+
+/** A graph node indicating a Array Node. */
+export interface ArrayNode extends Core.IArrayNode {
+  node: CompiledNode;
+  parallelism?: number;
+  minSuccesses?: number;
+  minSuccessRatio?: number;
+}
+
 /** A graph node indicating a task to be executed. This is the most common
  * node type in a Flyte graph.
  */
@@ -30,6 +39,7 @@ export interface CompiledNode extends Core.INode {
   inputs?: Binding[];
   metadata?: CompiledNodeMetadata;
   outputAliases?: Alias[];
+  arrayNode?: ArrayNode;
   taskNode?: TaskNode;
   upstreamNodeIds?: string[];
   workflowNode?: WorkflowNode;