diff --git a/tests/flytekit/unit/core/test_composition.py b/tests/flytekit/unit/core/test_composition.py index 28cbcdf706..3963c77c8d 100644 --- a/tests/flytekit/unit/core/test_composition.py +++ b/tests/flytekit/unit/core/test_composition.py @@ -35,7 +35,7 @@ def my_wf(a: int, b: str) -> (int, str, str): def test_single_named_output_subwf(): - nt = typing.NamedTuple("SubWfOutput", sub_int=int) + nt = NamedTuple("SubWfOutput", sub_int=int) @task def t1(a: int) -> nt: @@ -70,7 +70,7 @@ def my_wf(a: int) -> int: def test_lp_default_handling(): @task - def t1(a: int) -> typing.NamedTuple("OutputsBC", t1_int_output=int, c=str): + def t1(a: int) -> NamedTuple("OutputsBC", t1_int_output=int, c=str): a = a + 2 return a, "world-" + str(a)