lint

Author: opendansor

bittensor/commands/stake.py

@@ -609,7 +609,7 @@ def run(cli: "bittensor.cli"):
     @staticmethod
     def _run(cli: "bittensor.cli", subtensor: "bittensor.subtensor"):
         wallet = bittensor.wallet(config=cli.config)
-         
+
         # Get values if not set.
         if not cli.config.is_set("netuid"):
             cli.config.netuid = int(Prompt.ask("Enter netuid"))
@@ -618,7 +618,12 @@ def _run(cli: "bittensor.cli", subtensor: "bittensor.subtensor"):
             cli.config.hotkey = Prompt.ask("Enter parent hotkey (ss58)")
 
         # display children
-        GetChildrenCommand.retrieve_children(subtensor=subtensor, hotkey=cli.config.hotkey, netuid=cli.config.netuid, render_table=True)
+        GetChildrenCommand.retrieve_children(
+            subtensor=subtensor,
+            hotkey=cli.config.hotkey,
+            netuid=cli.config.netuid,
+            render_table=True,
+        )
 
         if not cli.config.is_set("children"):
             cli.config.children = Prompt.ask(
@@ -663,8 +668,12 @@ def _run(cli: "bittensor.cli", subtensor: "bittensor.subtensor"):
 
         # Result
         if success:
-            GetChildrenCommand.retrieve_children(subtensor=subtensor, hotkey=cli.config.hotkey,
-                                                 netuid=cli.config.netuid, render_table=True)
+            GetChildrenCommand.retrieve_children(
+                subtensor=subtensor,
+                hotkey=cli.config.hotkey,
+                netuid=cli.config.netuid,
+                render_table=True,
+            )
             console.print(
                 ":white_heavy_check_mark: [green]Set children hotkeys.[/green]"
             )
@@ -780,9 +789,11 @@ def _run(cli: "bittensor.cli", subtensor: "bittensor.subtensor"):
         GetChildrenCommand.render_table(subtensor, hotkey, children, netuid, True)
 
         return children
-    
+
     @staticmethod
-    def retrieve_children(subtensor: "bittensor.subtensor", hotkey: str, netuid: int, render_table: bool):
+    def retrieve_children(
+        subtensor: "bittensor.subtensor", hotkey: str, netuid: int, render_table: bool
+    ):
         children = subtensor.get_children(hotkey, netuid)
         if render_table:
             GetChildrenCommand.render_table(subtensor, hotkey, children, netuid, False)
@@ -834,15 +845,19 @@ def render_table(
 
         if not children:
             console.print(table)
-            console.print(f"There are currently no child hotkeys on subnet {netuid} with ParentHotKey {hotkey}.")
+            console.print(
+                f"There are currently no child hotkeys on subnet {netuid} with ParentHotKey {hotkey}."
+            )
             if prompt:
                 command = f"btcli stake set_children --children <child_hotkey> --hotkey <parent_hotkey> --netuid {netuid} --proportion <float>"
-                console.print(f"To add a child hotkey you can run the command: [white]{command}[/white]")
+                console.print(
+                    f"To add a child hotkey you can run the command: [white]{command}[/white]"
+                )
             return
 
         console.print("ParentHotKey:", style="cyan", no_wrap=True)
         console.print(hotkey)
-        
+
         # calculate totals
         total_proportion = 0
         total_stake = 0
@@ -867,7 +882,9 @@ def render_table(
 
         # add the children info to the table
         for i, (proportion, hotkey, stake) in enumerate(children_info, 1):
-            proportion_str = Text(str(proportion), style="red" if proportion == 0 else "")
+            proportion_str = Text(
+                str(proportion), style="red" if proportion == 0 else ""
+            )
             hotkey = Text(hotkey, style="red" if proportion == 0 else "")
             table.add_row(
                 str(i),
@@ -879,4 +896,3 @@ def render_table(
         # add totals row
         table.add_row("", "Total", str(total_proportion), str(total_stake), "")
         console.print(table)
-        

bittensor/commands/unstake.py

@@ -65,21 +65,21 @@ def check_config(cls, config: "bittensor.config"):
             config.wallet.name = str(wallet_name)
 
         if (
-                not config.get("hotkey_ss58address", d=None)
-                and not config.is_set("wallet.hotkey")
-                and not config.no_prompt
-                and not config.get("all_hotkeys")
-                and not config.get("hotkeys")
+            not config.get("hotkey_ss58address", d=None)
+            and not config.is_set("wallet.hotkey")
+            and not config.no_prompt
+            and not config.get("all_hotkeys")
+            and not config.get("hotkeys")
         ):
             hotkey = Prompt.ask("Enter hotkey name", default=defaults.wallet.hotkey)
             config.wallet.hotkey = str(hotkey)
 
         # Get amount.
         if (
-                not config.get("hotkey_ss58address")
-                and not config.get("amount")
-                and not config.get("unstake_all")
-                and not config.get("max_stake")
+            not config.get("hotkey_ss58address")
+            and not config.get("amount")
+            and not config.get("unstake_all")
+            and not config.get("max_stake")
         ):
             hotkeys: str = ""
             if config.get("all_hotkeys"):
@@ -270,13 +270,13 @@ def _run(cli: "bittensor.cli", subtensor: "bittensor.subtensor"):
         # Ask to unstake
         if not cli.config.no_prompt:
             if not Confirm.ask(
-                    f"Do you want to unstake from the following keys to {wallet.name}:\n"
-                    + "".join(
-                        [
-                            f"    [bold white]- {hotkey[0] + ':' if hotkey[0] else ''}{hotkey[1]}: {f'{amount} {bittensor.__tao_symbol__}' if amount else 'All'}[/bold white]\n"
-                            for hotkey, amount in zip(final_hotkeys, final_amounts)
-                        ]
-                    )
+                f"Do you want to unstake from the following keys to {wallet.name}:\n"
+                + "".join(
+                    [
+                        f"    [bold white]- {hotkey[0] + ':' if hotkey[0] else ''}{hotkey[1]}: {f'{amount} {bittensor.__tao_symbol__}' if amount else 'All'}[/bold white]\n"
+                        for hotkey, amount in zip(final_hotkeys, final_amounts)
+                    ]
+                )
             ):
                 return None
 
@@ -345,8 +345,12 @@ def _run(cli: "bittensor.cli", subtensor: "bittensor.subtensor"):
             cli.config.hotkey = Prompt.ask("Enter parent hotkey (ss58)")
 
         # Get and display current children information
-        current_children = GetChildrenCommand.retrieve_children(subtensor=subtensor, hotkey=cli.config.hotkey,
-                                                                netuid=cli.config.netuid, render_table=False)
+        current_children = GetChildrenCommand.retrieve_children(
+            subtensor=subtensor,
+            hotkey=cli.config.hotkey,
+            netuid=cli.config.netuid,
+            render_table=False,
+        )
 
         # Parse from strings
         netuid = cli.config.netuid
@@ -367,9 +371,12 @@ def _run(cli: "bittensor.cli", subtensor: "bittensor.subtensor"):
         # Result
         if success:
             if cli.config.wait_for_finalization and cli.config.wait_for_inclusion:
-                GetChildrenCommand.retrieve_children(subtensor=subtensor, hotkey=cli.config.hotkey,
-                                                     netuid=cli.config.netuid,
-                                                     render_table=True)
+                GetChildrenCommand.retrieve_children(
+                    subtensor=subtensor,
+                    hotkey=cli.config.hotkey,
+                    netuid=cli.config.netuid,
+                    render_table=True,
+                )
             console.print(
                 ":white_heavy_check_mark: [green]Revoked all children hotkeys.[/green]"
             )

bittensor/subtensor.py

@@ -3253,9 +3253,7 @@ def query_runtime_api(
         """
         call_definition = bittensor.__type_registry__["runtime_api"][runtime_api][  # type: ignore
             "methods"  # type: ignore
-        ][
-            method
-        ]  # type: ignore
+        ][method]  # type: ignore
 
         json_result = self.state_call(
             method=f"{runtime_api}_{method}",