diff --git a/blockchain/proof_of_stake.py b/blockchain/proof_of_stake.py index 78dc45c6f..1cb29ce1e 100644 --- a/blockchain/proof_of_stake.py +++ b/blockchain/proof_of_stake.py @@ -1,17 +1,7 @@ import random -from typing import List - class Validator: - """ - Represents a validator in a Proof of Stake system. - - Attributes: - name (str): The name of the validator. - stake (int): The amount of stake (coins) the validator holds. - """ - - def __init__(self, name: str, stake: int): + def __init__(self, name: str, stake: int) -> None: """ Initializes a new validator with a given name and stake. @@ -22,15 +12,14 @@ class Validator: self.name = name self.stake = stake - -def choose_validator(validators: List[Validator]) -> Validator: +def choose_validator(validators: list[Validator]) -> Validator: """ Selects a validator to create the next block based on the weight of their stake. The higher the stake, the greater the chance to be selected. Args: - validators (List[Validator]): A list of Validator objects. + validators (list[Validator]): A list of Validator objects. Returns: Validator: The selected validator based on weighted random selection. @@ -43,7 +32,6 @@ def choose_validator(validators: List[Validator]) -> Validator: """ total_stake = sum(v.stake for v in validators) weighted_validators = [(v, v.stake / total_stake) for v in validators] - selected = random.choices( - [v[0] for v in weighted_validators], weights=[v[1] for v in weighted_validators] - ) + selected = random.choices([v[0] for v in weighted_validators], + weights=[v[1] for v in weighted_validators]) return selected[0]