From afe41eb75cb7ec5d5edf54fe490bdde49294683b Mon Sep 17 00:00:00 2001
From: madt1m <pietrotirenna.pt@gmail.com>
Date: Wed, 1 Aug 2018 12:00:07 +0200
Subject: protobuf: changed return type annotation in loads to enhance
 granularity

---
 mitmproxy/io/protobuf.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mitmproxy/io/protobuf.py b/mitmproxy/io/protobuf.py
index 9a00eacf..c8ca3acc 100644
--- a/mitmproxy/io/protobuf.py
+++ b/mitmproxy/io/protobuf.py
@@ -189,7 +189,7 @@ def load_http(hf: http_pb2.HTTPFlow) -> HTTPFlow:
     return f
 
 
-def loads(b: bytes, typ="http") -> flow.Flow:
+def loads(b: bytes, typ="http") -> typing.Union[HTTPFlow]:
     if typ != 'http':
         raise exceptions.TypeError("Flow types different than HTTP not supported yet!")
     else:
-- 
cgit v1.2.3