From 42583eaceb9f32b104c39edb81b849d4580371be Mon Sep 17 00:00:00 2001
From: matejcik <ja@matejcik.cz>
Date: Fri, 1 Feb 2019 14:27:52 +0100
Subject: [PATCH] debuglink: allow clearing input_flow function

---
 trezorlib/debuglink.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/trezorlib/debuglink.py b/trezorlib/debuglink.py
index de2609abf..9a94bfbb1 100644
--- a/trezorlib/debuglink.py
+++ b/trezorlib/debuglink.py
@@ -240,6 +240,10 @@ class TrezorClientDebugLink(TrezorClient):
             return msg
 
     def set_input_flow(self, input_flow):
+        if input_flow is None:
+            self.ui.input_flow = None
+            return
+
         if callable(input_flow):
             input_flow = input_flow()
         if not hasattr(input_flow, "send"):