use in dict

Signed-off-by: Yee Hing Tong <[email protected]>

flytekit/core/type_engine.py

@@ -1566,9 +1566,7 @@ async def async_to_literal(
                 asyncio.create_task(TypeEngine.async_to_literal(ctx, x, t, expected.collection_type))
                 for x in python_val
             ]
-            print("About to run!")
-            with timeit("run coros"):
-                lit_list = await _run_coros_in_chunks(lit_list)
+            lit_list = await _run_coros_in_chunks(lit_list)
 
         return Literal(collection=LiteralCollection(literals=lit_list))
 
@@ -1599,7 +1597,7 @@ async def async_to_python_value(  # type: ignore
         else:
             st = self.get_sub_type(expected_python_type)
             result = [TypeEngine.async_to_python_value(ctx, x, st) for x in lits]
-            result = await asyncio.gather(*result)
+            result = await _run_coros_in_chunks(result)
             return result  # type: ignore  # should be a list, thinks its a tuple
 
     def guess_python_type(self, literal_type: LiteralType) -> list:  # type: ignore
@@ -2005,7 +2003,7 @@ async def async_to_literal(
                 TypeEngine.async_to_literal(ctx, v, cast(type, v_type), expected.map_value_type)
             )
 
-        await asyncio.gather(*lit_map.values())
+        await _run_coros_in_chunks([c for c in lit_map.values()])
         for k, v in lit_map.items():
             lit_map[k] = v.result()
 
@@ -2031,7 +2029,7 @@ async def async_to_python_value(self, ctx: FlyteContext, lv: Literal, expected_p
                 fut = asyncio.create_task(TypeEngine.async_to_python_value(ctx, v, cast(Type, tp[1])))
                 py_map[k] = fut
 
-            await asyncio.gather(*py_map.values())
+            await _run_coros_in_chunks([c for c in py_map.values()])
             for k, v in py_map.items():
                 py_map[k] = v.result()