diff --git a/sw_utils/protocol_config.py b/sw_utils/protocol_config.py index df04479..f0adbb7 100644 --- a/sw_utils/protocol_config.py +++ b/sw_utils/protocol_config.py @@ -2,7 +2,7 @@ def build_protocol_config( - config_data: dict, rewards_threshold: int = 0, validators_threshold: int = 0 + config_data: dict, rewards_threshold: int | None = None, validators_threshold: int | None = None ) -> ProtocolConfig: oracles = [] for oracle in config_data['oracles']: @@ -13,10 +13,10 @@ def build_protocol_config( ) ) - if rewards_threshold and not 1 <= rewards_threshold <= len(oracles): + if rewards_threshold is not None and not 1 <= rewards_threshold <= len(oracles): raise ValueError('Invalid rewards threshold') - if validators_threshold and not 1 <= validators_threshold <= len(oracles): + if validators_threshold is not None and not 1 <= validators_threshold <= len(oracles): raise ValueError('Invalid validators threshold') public_keys = [oracle.public_key for oracle in oracles] @@ -30,8 +30,8 @@ def build_protocol_config( return ProtocolConfig( oracles=oracles, - rewards_threshold=rewards_threshold, - validators_threshold=validators_threshold, + rewards_threshold=rewards_threshold or 0, + validators_threshold=validators_threshold or 0, exit_signature_recover_threshold=exit_signature_recover_threshold, supported_relays=config_data['supported_relays'], vault_fee_max_bps=config_data['vault_max_fee'],