move cluster pool arg (#91)

Signed-off-by: squiishyy <[email protected]>
flyteorg · Oct 26, 2023 · d19a2cf · d19a2cf
1 parent 9e95508
commit d19a2cf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/boilerplate/flyte/end2end/run-tests.py b/boilerplate/flyte/end2end/run-tests.py
@@ -97,8 +97,8 @@ def execute_workflow(
         cluster_pool_name: Optional[str] = None,
 ):
     print(f"Fetching workflow={workflow_name} and version={version}")
-    wf = remote.fetch_workflow(name=workflow_name, version=version, cluster_pool=cluster_pool_name)
-    return remote.execute(wf, inputs=inputs, wait=False)
+    wf = remote.fetch_workflow(name=workflow_name, version=version)
+    return remote.execute(wf, inputs=inputs, wait=False, cluster_pool=cluster_pool_name)
 
 
 def executions_finished(executions_by_wfgroup: Dict[str, List[FlyteWorkflowExecution]]) -> bool: