Ruff

Author: ibraheem-abe

bittensor/core/async_subtensor.py

@@ -2831,14 +2831,14 @@ async def add_stake(
             allow_partial_stake (bool): If true and safe_staking is enabled, allows partial staking when
                 the full amount would exceed the price threshold. If false, the entire stake fails if it would
                 exceed the threshold. Default is False.
-            rate_threshold (float): The maximum allowed price change ratio when staking. For example, 
+            rate_threshold (float): The maximum allowed price change ratio when staking. For example,
                 0.005 = 0.5% maximum price increase. Only used when safe_staking is True. Default is 0.005.
 
         Returns:
             bool: ``True`` if the staking is successful, False otherwise.
 
-        This function enables neurons to increase their stake in the network, enhancing their influence and potential
-            rWhen safe_staking is enabled, it provides protection against price fluctuations during the time stake is
+        This function enables neurons to increase their stake in the network, enhancing their influence and potential.
+            When safe_staking is enabled, it provides protection against price fluctuations during the time stake is
             executed and the time it is actually processed by the chain.
         """
         amount = check_and_convert_to_balance(amount)
@@ -3632,7 +3632,7 @@ async def unstake(
             allow_partial_stake (bool): If true and safe_staking is enabled, allows partial unstaking when
                 the full amount would exceed the price threshold. If false, the entire unstake fails if it would
                 exceed the threshold. Default is False.
-            rate_threshold (float): The maximum allowed price change ratio when unstaking. For example, 
+            rate_threshold (float): The maximum allowed price change ratio when unstaking. For example,
                 0.005 = 0.5% maximum price decrease. Only used when safe_staking is True. Default is 0.005.
 
         Returns:

tests/e2e_tests/test_staking.py

@@ -485,7 +485,9 @@ def test_safe_swap_stake_scenarios(subtensor, alice_wallet, bob_wallet):
         alice_wallet.hotkey.ss58_address,
         netuid=dest_netuid,
     )
-    assert dest_stake == Balance(0), "Destination stake should remain 0 after failed swap"
+    assert dest_stake == Balance(
+        0
+    ), "Destination stake should remain 0 after failed swap"
 
     # 2. Try swap with higher threshold - should succeed
     success = subtensor.swap_stake(
@@ -513,4 +515,6 @@ def test_safe_swap_stake_scenarios(subtensor, alice_wallet, bob_wallet):
         alice_wallet.hotkey.ss58_address,
         netuid=dest_netuid,
     )
-    assert dest_stake > Balance(0), "Destination stake should be non-zero after successful swap"
+    assert dest_stake > Balance(
+        0
+    ), "Destination stake should be non-zero after successful swap"