From 38af43b7ec6765a880e74f0f3d71f513d2061965 Mon Sep 17 00:00:00 2001 From: Chih Tsung Lu Date: Wed, 5 Feb 2025 15:52:46 +0800 Subject: [PATCH] fix: map implementation in test_array_node_map_task.py Signed-off-by: lu00122 Signed-off-by: Chih Tsung Lu --- tests/flytekit/unit/core/test_array_node_map_task.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/flytekit/unit/core/test_array_node_map_task.py b/tests/flytekit/unit/core/test_array_node_map_task.py index c4329041fd..f2d5b927c0 100644 --- a/tests/flytekit/unit/core/test_array_node_map_task.py +++ b/tests/flytekit/unit/core/test_array_node_map_task.py @@ -9,7 +9,7 @@ import pytest from flyteidl.core import workflow_pb2 as _core_workflow -from flytekit import dynamic, map_task, task, workflow, eager, PythonFunctionTask, Resources +from flytekit import dynamic, map, task, workflow, eager, PythonFunctionTask, Resources from flytekit.configuration import FastSerializationSettings, Image, ImageConfig, SerializationSettings from flytekit.core import context_manager from flytekit.core.array_node_map_task import ArrayNodeMapTask, ArrayNodeMapTaskResolver