diff --git a/pyproject.toml b/pyproject.toml index 42ea0367d..254cb51dd 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -28,6 +28,7 @@ dependencies = [ "scalecodec==1.2.11", "typer>=0.16", "bittensor-wallet>=3.0.7", + "packaging", "plotille>=5.0.0", "plotly>=6.0.0", ] diff --git a/tests/e2e_tests/utils.py b/tests/e2e_tests/utils.py index a2a87e48b..b8b729b3e 100644 --- a/tests/e2e_tests/utils.py +++ b/tests/e2e_tests/utils.py @@ -1,3 +1,4 @@ +import importlib import inspect import os import re @@ -6,10 +7,12 @@ import sys from typing import TYPE_CHECKING, Optional -from bittensor_cli.cli import CLIManager from bittensor_wallet import Keypair, Wallet +from packaging.version import parse as parse_version, Version from typer.testing import CliRunner +from bittensor_cli.cli import CLIManager + if TYPE_CHECKING: from async_substrate_interface.async_substrate import AsyncSubstrateInterface @@ -55,7 +58,10 @@ def exec_command( extra_args.extend(["--network", "ws://127.0.0.1:9945"]) # Capture stderr separately from stdout - runner = CliRunner() + if parse_version(importlib.metadata.version("click")) < Version("8.2.0"): + runner = CliRunner(mix_stderr=False) + else: + runner = CliRunner() # Prepare the command arguments args = [ command,