From e6f615e9511b15168c83ccf170ba01baa200089b Mon Sep 17 00:00:00 2001 From: Yicheng-Lu-llll Date: Mon, 13 Mar 2023 04:38:22 +0000 Subject: [PATCH] improve format Signed-off-by: Yicheng-Lu-llll --- tests/flytekit/unit/core/test_type_engine.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/flytekit/unit/core/test_type_engine.py b/tests/flytekit/unit/core/test_type_engine.py index f04b61651d..90d2c5510c 100644 --- a/tests/flytekit/unit/core/test_type_engine.py +++ b/tests/flytekit/unit/core/test_type_engine.py @@ -1580,8 +1580,11 @@ def test_file_ext_with_flyte_file_wrong_type(): "python_val, python_type, batch_size", [ ([{"a": {0: "foo"}}] * 5, typing.List[typing.Dict[str, FlytePickle]], 5), - ([{"a": {0: "foo"}}] * 5, Annotated[typing.List[typing.Dict[str, FlytePickle]], 2], 2), - ([{"a": {0: "foo"}}] * 6, Annotated[typing.List[typing.Dict[str, FlytePickle]], HashMethod(function=str), 2], 2), + ( + [{"a": {0: "foo"}}] * 5, + Annotated[typing.List[typing.Dict[str, FlytePickle]], HashMethod(function=str), 2], + 2, + ), ], ) def test_batch_pickle_list(python_val, python_type, batch_size):