support structure dataclass and flytescheme with DataClassJSONMixin

Signed-off-by: hhcs9527 <hhcs9527@gmail.com>

flytekit/types/schema/types.py

@@ -19,6 +19,7 @@
 from flytekit.loggers import logger
 from flytekit.models.literals import Literal, Scalar, Schema
 from flytekit.models.types import LiteralType, SchemaType
+from mashumaro.mixins.json import DataClassJSONMixin
 
 T = typing.TypeVar("T")
 
@@ -179,9 +180,8 @@ def get_handler(cls, t: Type) -> SchemaHandler:
         return cls._SCHEMA_HANDLERS[t]
 
 
-@dataclass_json
 @dataclass
-class FlyteSchema(object):
+class FlyteSchema(DataClassJSONMixin):
     remote_path: typing.Optional[str] = field(default=None, metadata=config(mm_field=fields.String()))
     """
     This is the main schema class that users should use.

flytekit/types/structured/structured_dataset.py

@@ -22,6 +22,7 @@
 from flytekit.models import types as type_models
 from flytekit.models.literals import Literal, Scalar, StructuredDatasetMetadata
 from flytekit.models.types import LiteralType, SchemaType, StructuredDatasetType
+from mashumaro.mixins.json import DataClassJSONMixin
 
 if typing.TYPE_CHECKING:
     import pandas as pd
@@ -43,9 +44,8 @@
 GENERIC_PROTOCOL: str = "generic protocol"
 
 
-@dataclass_json
 @dataclass
-class StructuredDataset(object):
+class StructuredDataset(DataClassJSONMixin):
     """
     This is the user facing StructuredDataset class. Please don't confuse it with the literals.StructuredDataset
     class (that is just a model, a Python class representation of the protobuf).