run black

opentensor · Aug 23, 2023 · 9af0a3b · 9af0a3b
1 parent 1c2fd89
commit 9af0a3b
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 15 deletions.
diff --git a/bittensor/chain_data.py b/bittensor/chain_data.py
@@ -716,7 +716,8 @@ def delegated_list_from_vec_u8(
         ]
 
         return decoded
-
+
+
 @dataclass
 class StakeInfo:
     r"""
@@ -731,13 +732,9 @@ def fix_decoded_values(cls, decoded: Any) -> "StakeInfo":
         r"""Fixes the decoded values."""
 
         return cls(
-            hotkey_ss58=ss58_encode(
-                decoded["hotkey"], bittensor.__ss58_format__
-            ),
+            hotkey_ss58=ss58_encode(decoded["hotkey"], bittensor.__ss58_format__),
             coldkey_ss58=ss58_encode(decoded["coldkey"], bittensor.__ss58_format__),
-            stake=Balance.from_rao(
-                decoded["stake"]
-            ),
+            stake=Balance.from_rao(decoded["stake"]),
         )
 
     @classmethod

diff --git a/bittensor/subtensor.py b/bittensor/subtensor.py
@@ -1792,7 +1792,7 @@ def get_stake_info_for_colkey(
     ) -> List[Tuple[DelegateInfo, Balance]]:
         """Returns the list of StakeInfo objects for this coldkey"""
 
-        encoded_coldkey = ss58_to_vec_u8(coldkey_ss58)        
+        encoded_coldkey = ss58_to_vec_u8(coldkey_ss58)
 
         hex_bytes_result = self.query_runtime_api(
             runtime_api="StakeInfoRuntimeApi",
@@ -1803,19 +1803,21 @@ def get_stake_info_for_colkey(
 
         if hex_bytes_result == None:
             return None
-        
+
         if hex_bytes_result.startswith("0x"):
             bytes_result = bytes.fromhex(hex_bytes_result[2:])
         else:
             bytes_result = bytes.fromhex(hex_bytes_result)
-            
+
         return StakeInfo.list_from_vec_u8(bytes_result)
-    
+
     def get_stake_info_for_colkeys(
         self, coldkey_ss58_list: List[str], block: Optional[int] = None
     ) -> List[Tuple[DelegateInfo, Balance]]:
         """Returns the list of StakeInfo objects for all coldkeys in the list."""
-        encoded_coldkeys = [ss58_to_vec_u8(coldkey_ss58) for coldkey_ss58 in coldkey_ss58_list]
+        encoded_coldkeys = [
+            ss58_to_vec_u8(coldkey_ss58) for coldkey_ss58 in coldkey_ss58_list
+        ]
 
         hex_bytes_result = self.query_runtime_api(
             runtime_api="StakeInfoRuntimeApi",
@@ -1826,14 +1828,14 @@ def get_stake_info_for_colkeys(
 
         if hex_bytes_result == None:
             return None
-        
+
         if hex_bytes_result.startswith("0x"):
             bytes_result = bytes.fromhex(hex_bytes_result[2:])
         else:
             bytes_result = bytes.fromhex(hex_bytes_result)
-            
+
         return StakeInfo.list_from_vec_u8(bytes_result)
-    
+
     ########################################
     #### Neuron information per subnet ####
     ########################################

diff --git a/bittensor/utils/__init__.py b/bittensor/utils/__init__.py
@@ -31,11 +31,13 @@
 U16_MAX = 65535
 U64_MAX = 18446744073709551615
 
+
 def ss58_to_vec_u8(ss58_address: str) -> List[int]:
     ss58_bytes: bytes = bittensor.utils.ss58_address_to_bytes(ss58_address)
     encoded_address: List[int] = [int(byte) for byte in ss58_bytes]
     return encoded_address
 
+
 def unbiased_topk(values, k, dim=0, sorted=True, largest=True):
     r"""Selects topk as in torch.topk but does not bias lower indices when values are equal.
     Args: