From e825e3881eb7b7b0480dc18580181284a7937b3c Mon Sep 17 00:00:00 2001 From: Kevin Su Date: Thu, 16 Feb 2023 00:51:44 -0800 Subject: [PATCH] update test Signed-off-by: Kevin Su --- tests/flytekit/unit/core/test_type_engine.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/flytekit/unit/core/test_type_engine.py b/tests/flytekit/unit/core/test_type_engine.py index 4e72fa0e79..997000bc5a 100644 --- a/tests/flytekit/unit/core/test_type_engine.py +++ b/tests/flytekit/unit/core/test_type_engine.py @@ -755,8 +755,8 @@ def test_structured_dataset_in_dataclass(): @dataclass class InnerDatasetStruct(object): a: People - b: typing.List[People] - c: typing.Dict[str, People] + b: typing.List[Annotated[StructuredDataset, "parquet"]] + c: typing.Dict[str, Annotated[StructuredDataset, kwtypes(Name=str, Age=int)]] @dataclass_json @dataclass