Sub in Sub IT

Signed-off-by: Hongxin Liang <[email protected]>
flyteorg · Oct 9, 2023 · 572bd7d · 572bd7d
1 parent 2d1d14e
commit 572bd7d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/flytekit-examples/src/main/java/org/flyte/examples/DynamicFibonacciWorkflowTask.java b/flytekit-examples/src/main/java/org/flyte/examples/DynamicFibonacciWorkflowTask.java
@@ -87,13 +87,18 @@ public Output run(SdkWorkflowBuilder builder, Input input) {
                       SdkTypes.nulls(),
                       SdkTypes.nulls())
                   .withUpstreamNode(hello));
+      SdkNode<WelcomeWorkflow.Output> greet =
+          builder.apply(
+              "greet",
+              new SubWorkflow().withUpstreamNode(world),
+              WelcomeWorkflow.Input.create(SdkBindingDataFactory.of("greet")));
       @Var SdkBindingData<Long> prev = SdkBindingDataFactory.of(0);
       @Var SdkBindingData<Long> value = SdkBindingDataFactory.of(1);
       for (int i = 2; i <= input.n().get(); i++) {
         SdkBindingData<Long> next =
             builder
                 .apply(
-                    "fib-" + i, new SumTask().withUpstreamNode(world), SumInput.create(value, prev))
+                    "fib-" + i, new SumTask().withUpstreamNode(greet), SumInput.create(value, prev))
                 .getOutputs();
         prev = value;
         value = next;