also update execution state for default case

Signed-off-by: oliverhu <[email protected]>
flyteorg · Jul 24, 2023 · 656cf30 · 656cf30
1 parent 529b6cf
commit 656cf30
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/flytekit/core/promise.py b/flytekit/core/promise.py
@@ -1028,9 +1028,10 @@ def flyte_entity_call_handler(
                     return None
             return cast(LocallyExecutable, entity).local_execute(child_ctx, **kwargs)
     else:
+        mode = cast(LocallyExecutable, entity).local_execution_mode()
         with FlyteContextManager.with_context(
             ctx.with_execution_state(
-                ctx.new_execution_state().with_params(mode=ExecutionState.Mode.LOCAL_WORKFLOW_EXECUTION)
+                ctx.new_execution_state().with_params(mode=mode)
             )
         ) as child_ctx:
             cast(ExecutionParameters, child_ctx.user_space_params)._decks = []