From 127e21f3eb85851219003e6279290d7863888465 Mon Sep 17 00:00:00 2001 From: Yee Hing Tong Date: Mon, 4 Nov 2024 16:56:52 -0800 Subject: [PATCH] Map/setup exec (#2898) Signed-off-by: Yee Hing Tong --- flytekit/bin/entrypoint.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flytekit/bin/entrypoint.py b/flytekit/bin/entrypoint.py index fc9e16014a..2eed76321f 100644 --- a/flytekit/bin/entrypoint.py +++ b/flytekit/bin/entrypoint.py @@ -476,7 +476,7 @@ def _execute_map_task( raise ValueError(f"Resolver args cannot be <1, got {resolver_args}") with setup_execution( - raw_output_data_prefix, checkpoint_path, prev_checkpoint, dynamic_addl_distro, dynamic_dest_dir + raw_output_data_prefix, output_prefix, checkpoint_path, prev_checkpoint, dynamic_addl_distro, dynamic_dest_dir ) as ctx: working_dir = os.getcwd() if all(os.path.realpath(path) != working_dir for path in sys.path):