[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2024-10-10 15:54:30 +00:00
parent 28df0f3c9b
commit d770fa5bf5

View File

@ -1,6 +1,7 @@
import random
from typing import List
class Validator:
"""
Represents a validator in a Proof of Stake system.
@ -9,6 +10,7 @@ class Validator:
name (str): The name of the validator.
stake (int): The amount of stake (coins) the validator holds.
"""
def __init__(self, name: str, stake: int):
"""
Initializes a new validator with a given name and stake.
@ -20,6 +22,7 @@ class Validator:
self.name = name
self.stake = stake
def choose_validator(validators: List[Validator]) -> Validator:
"""
Selects a validator to create the next block based on the weight of their stake.
@ -40,6 +43,7 @@ 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]