update test

Signed-off-by: Kevin Su <[email protected]>
flyteorg · Feb 16, 2023 · e825e38 · e825e38
1 parent 237c0d5
commit e825e38
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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