Fix tests

Signed-off-by: Kevin Su <[email protected]>
flyteorg · Jun 16, 2022 · 6740aff · 6740aff
1 parent 33764df
commit 6740aff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)