From 096b8efa68aec823cc5884948f09e7e50cd86bc6 Mon Sep 17 00:00:00 2001 From: Samhita Alla Date: Mon, 11 Jul 2022 12:12:11 +0530 Subject: [PATCH] fix low codecov Signed-off-by: Samhita Alla --- tests/flytekit/unit/core/test_type_hints.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/flytekit/unit/core/test_type_hints.py b/tests/flytekit/unit/core/test_type_hints.py index df07c03a62..abdf69f5b0 100644 --- a/tests/flytekit/unit/core/test_type_hints.py +++ b/tests/flytekit/unit/core/test_type_hints.py @@ -91,9 +91,10 @@ def my_task(a: int) -> typing.NamedTuple("OutputsBC", b=typing.ForwardRef("int") def test_annotated_namedtuple_output(): @task - def my_task() -> typing.NamedTuple("OutputA", a=Annotated[int, "metadata-a"]): - return 1 + def my_task(a: int) -> typing.NamedTuple("OutputA", a=Annotated[int, "metadata-a"]): + return a + 2 + assert my_task(a=9) == (11,) assert get_origin(my_task.python_interface.outputs["a"]) is Annotated