diff --git a/starknet_py/net/schemas/rpc/contract.py b/starknet_py/net/schemas/rpc/contract.py index f774a4b02..7cf3f8a21 100644 --- a/starknet_py/net/schemas/rpc/contract.py +++ b/starknet_py/net/schemas/rpc/contract.py @@ -2,7 +2,7 @@ from marshmallow import EXCLUDE from marshmallow import Schema as MarshmallowSchema -from marshmallow import SchemaOpts, ValidationError, fields, post_load +from marshmallow import ValidationError, fields, post_load from starknet_py.abi.v0.schemas import ContractAbiEntrySchema from starknet_py.net.client_models import ( @@ -22,7 +22,7 @@ from starknet_py.net.executable_models import CasmClass from starknet_py.net.schemas.common import Felt, NumberAsHex from starknet_py.net.schemas.rpc.executables_api import HintSchema -from starknet_py.utils.schema import ExcludeOpts, Schema +from starknet_py.utils.schema import Schema class SyncStatusSchema(Schema): @@ -183,10 +183,7 @@ def make_dataclass(self, data, **kwargs) -> CasmClassEntryPointsByType: return CasmClassEntryPointsByType(**data) -# TODO(#1564): `CasmClassSchema` should inherit from `Schema` and shouldn't overwrite `OPTION_CLASS` -# once issue is resolved. class CasmClassSchema(MarshmallowSchema): - OPTIONS_CLASS = ExcludeOpts prime = NumberAsHex(data_key="prime", required=True) bytecode = fields.List(Felt(), data_key="bytecode", required=True) bytecode_segment_lengths = fields.List( diff --git a/starknet_py/utils/schema.py b/starknet_py/utils/schema.py index 7f8ffb051..304bbd214 100644 --- a/starknet_py/utils/schema.py +++ b/starknet_py/utils/schema.py @@ -16,13 +16,5 @@ def __init__(self, meta, **kwargs): ) -# TODO(#1564): `ExcludeOpts` should be removed once issue is resolved. -class ExcludeOpts(SchemaOpts): - - def __init__(self, meta, **kwargs): - SchemaOpts.__init__(self, meta, **kwargs) - self.unknown = EXCLUDE - - class Schema(MarshmallowSchema): OPTIONS_CLASS = UnknownOpts