Skip to content

Commit

Permalink
improve format
Browse files Browse the repository at this point in the history
Signed-off-by: Yicheng-Lu-llll <[email protected]>
  • Loading branch information
Yicheng-Lu-llll committed Mar 13, 2023
1 parent 01aba67 commit e6f615e
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions tests/flytekit/unit/core/test_type_engine.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit e6f615e

Please sign in to comment.