Changes to proof_of_stake.py

Author: divyaSree-S0

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 @@ def __init__(self, name: str, stake: int):
         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]