From 82064e3a23bbb1fd7dcb115ae255bbc011de17cf Mon Sep 17 00:00:00 2001 From: DIVYASREE S Date: Thu, 10 Oct 2024 21:32:04 +0530 Subject: [PATCH] Changes to proof_of_stake.py --- blockchain/proof_of_stake.py | 22 +++++----------------- 1 file changed, 5 insertions(+), 17 deletions(-) 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]