diff --git a/pyproject.toml b/pyproject.toml index 3d504d0..192c99b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "vana" -version = "0.35.0" +version = "0.36.0" description = "" authors = ["Tim Nunamaker ", "Volodymyr Isai ", "Kahtaf Alam "] readme = "README.md" diff --git a/vana/__init__.py b/vana/__init__.py index 75377f5..c87fe42 100644 --- a/vana/__init__.py +++ b/vana/__init__.py @@ -15,7 +15,7 @@ # OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER # DEALINGS IN THE SOFTWARE. -__version__ = "0.35.0" +__version__ = "0.36.0" import rich diff --git a/vana/commands/satya.py b/vana/commands/satya.py index 0c14143..04a790f 100644 --- a/vana/commands/satya.py +++ b/vana/commands/satya.py @@ -53,10 +53,11 @@ def run(cli: "vana.cli"): wallet = vana.Wallet(config=cli.config if cli.config.wallet else None) tee_address = cli.config.get('tee_address', wallet.hotkey.address) + public_key = cli.config.get('public_key', wallet.get_hotkey_public_key()) tx_hash, tx_receipt = vana_client.register_tee( url=cli.config.url, - public_key=wallet.get_hotkey_public_key(), + public_key=public_key, tee_address=tee_address ) @@ -65,7 +66,7 @@ def run(cli: "vana.cli"): f"[bold green]Successfully registered validator node with:" f"\n- URL: '{cli.config.url}'" f"\n- TEE Address: '{tee_address}'" - f"\n- Public Key: '{wallet.get_hotkey_public_key()}'[/bold green]" + f"\n- Public Key: '{public_key}'[/bold green]" ) vana.__console__.print(f"Transaction hash: {tx_hash.hex()}") else: @@ -88,6 +89,8 @@ def add_args(parser: argparse.ArgumentParser): help="The network to use for registration.") satya_parser.add_argument("--tee_address", type=str, required=False, help="The hotkey address of the TEE node to register.") + satya_parser.add_argument("--public_key", type=str, required=False, + help="The public key of the TEE node to register.") @staticmethod def check_config(config: "vana.Config"):