diff --git a/python/src/trezorlib/cli/trezorctl.py b/python/src/trezorlib/cli/trezorctl.py index 1269092941..5e6327e023 100755 --- a/python/src/trezorlib/cli/trezorctl.py +++ b/python/src/trezorlib/cli/trezorctl.py @@ -53,8 +53,8 @@ from . import ( stellar, tezos, with_client, - with_session, with_default_session, + with_session, ) F = TypeVar("F", bound=Callable) diff --git a/python/src/trezorlib/client.py b/python/src/trezorlib/client.py index 266ca9f8a8..8d81b6715e 100644 --- a/python/src/trezorlib/client.py +++ b/python/src/trezorlib/client.py @@ -52,6 +52,7 @@ UNKNOWN = -1 PROTOCOL_V1 = 1 PROTOCOL_V2 = 2 + class TrezorClient: button_callback: t.Callable[[Session, t.Any], t.Any] | None = None passphrase_callback: t.Callable[[Session, t.Any], t.Any] | None = None @@ -60,6 +61,7 @@ class TrezorClient: _management_session: Session | None = None _features: messages.Features | None = None _protocol_version: int + def __init__( self, transport: Transport,