Use the rewritten nodes

Signed-off-by: Hongxin Liang <[email protected]>
flyteorg · Oct 9, 2023 · edd6124 · edd6124
1 parent ab0ceeb
commit edd6124
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/jflyte-utils/src/main/java/org/flyte/jflyte/utils/ProjectClosure.java b/jflyte-utils/src/main/java/org/flyte/jflyte/utils/ProjectClosure.java
@@ -384,10 +384,10 @@ public static Map<WorkflowIdentifier, WorkflowTemplate> collectSubWorkflows(
   }
 
   public static Map<TaskIdentifier, TaskTemplate> collectDynamicWorkflowTasks(
-      List<Node> rewrittenNodes,
+      List<Node> nodes,
       Map<TaskIdentifier, TaskTemplate> allTasks,
       Function<TaskIdentifier, TaskTemplate> remoteTaskTemplateFetcher) {
-    return collectTaskIds(rewrittenNodes).stream()
+    return collectTaskIds(nodes).stream()
         // all identifiers should be rewritten at this point
         .map(
             taskId ->

diff --git a/jflyte/src/main/java/org/flyte/jflyte/ExecuteDynamicWorkflow.java b/jflyte/src/main/java/org/flyte/jflyte/ExecuteDynamicWorkflow.java
@@ -309,7 +309,7 @@ private static List<Node> collectTaskTemplates(
 
     Map<TaskIdentifier, TaskTemplate> usedTaskTemplates =
         ProjectClosure.collectDynamicWorkflowTasks(
-            nodes, allTaskTemplates, id -> fetchTaskTemplate(flyteAdminClient, id, cache));
+            rewrittenNodes, allTaskTemplates, id -> fetchTaskTemplate(flyteAdminClient, id, cache));
     allUsedTaskTemplates.putAll(usedTaskTemplates);
 
     return rewrittenNodes;