Merge branch 'array-node-input-transformation' of github.com:flyteorg…

…/flytekit into array-node-input-transformation
flyteorg · Jun 17, 2024 · 63a8ee7 · 63a8ee7
2 parents 05d2033 + 81abd66
commit 63a8ee7
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/flytekit/core/array_node_map_task.py b/flytekit/core/array_node_map_task.py
@@ -238,13 +238,10 @@ def _literal_map_to_python_input(
     def execute(self, **kwargs) -> Any:
         ctx = FlyteContextManager.current_context()
         if ctx.execution_state and ctx.execution_state.mode == ExecutionState.Mode.TASK_EXECUTION:
-            return self._execute_map_task(**kwargs)
+            return exception_scopes.user_entry_point(self.python_function_task.execute)(**kwargs)
 
         return self._raw_execute(**kwargs)
 
-    def _execute_map_task(self, _: FlyteContext, **kwargs) -> Any:
-        return exception_scopes.user_entry_point(self.python_function_task.execute)(**kwargs)
-
     @staticmethod
     def _compute_array_job_index() -> int:
         """