From e68a4e83b54ba7aa670ee43dd1e0834d9c846738 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fabio=20Gr=C3=A4tz?= Date: Fri, 9 Feb 2024 19:23:44 +0100 Subject: [PATCH] Adapt tests MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Fabio Grätz --- tests/flytekit/unit/core/test_array_node_map_task.py | 2 +- tests/flytekit/unit/core/test_map_task.py | 2 +- tests/flytekit/unit/core/test_node_creation.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) 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 40bb864c4f..4bcecde6a7 100644 --- a/tests/flytekit/unit/core/test_array_node_map_task.py +++ b/tests/flytekit/unit/core/test_array_node_map_task.py @@ -305,4 +305,4 @@ def my_mappable_task(a: int) -> typing.Optional[str]: def wf(x: typing.List[int]): array_node_map_task(my_mappable_task)(a=x).with_overrides(container_image="random:image") - assert wf.nodes[0].run_entity.container_image == "random:image" + assert wf.nodes[0]._container_image == "random:image" diff --git a/tests/flytekit/unit/core/test_map_task.py b/tests/flytekit/unit/core/test_map_task.py index c87d4c6b1f..2ae716d4b7 100644 --- a/tests/flytekit/unit/core/test_map_task.py +++ b/tests/flytekit/unit/core/test_map_task.py @@ -352,7 +352,7 @@ def my_mappable_task(a: int) -> typing.Optional[str]: def wf(x: typing.List[int]): map_task(my_mappable_task)(a=x).with_overrides(container_image="random:image") - assert wf.nodes[0].flyte_entity.run_task.container_image == "random:image" + assert wf.nodes[0]._container_image == "random:image" def test_bounded_inputs_vars_order(serialization_settings): diff --git a/tests/flytekit/unit/core/test_node_creation.py b/tests/flytekit/unit/core/test_node_creation.py index df16ddd244..56eb82aa1d 100644 --- a/tests/flytekit/unit/core/test_node_creation.py +++ b/tests/flytekit/unit/core/test_node_creation.py @@ -465,7 +465,7 @@ def wf() -> str: bar().with_overrides(container_image="hello/world") return "hi" - assert wf.nodes[0].flyte_entity.container_image == "hello/world" + assert wf.nodes[0]._container_image == "hello/world" def test_override_accelerator():