Fix mypy errors (#504)

abhinavsingh · Feb 21, 2021 · 864a0b8 · 864a0b8
1 parent 20fb794
commit 864a0b8
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/proxy/core/ssh/tunnel.py b/proxy/core/ssh/tunnel.py
@@ -8,7 +8,7 @@
     :copyright: (c) 2013-present by Abhinav Singh and contributors.
     :license: BSD, see LICENSE for more details.
 """
-from typing import Tuple, Callable
+from typing import Optional, Tuple, Callable
 
 import paramiko
 
@@ -44,11 +44,15 @@ def run(self) -> None:
                 key_filename=self.private_pem_key
             )
             print('SSH connection established...')
-            transport: paramiko.transport.Transport = ssh.get_transport()
+            transport: Optional[paramiko.transport.Transport] = ssh.get_transport(
+            )
+            assert transport is not None
             transport.request_port_forward('', self.remote_proxy_port)
             print('Tunnel port forward setup successful...')
             while True:
-                conn: paramiko.channel.Channel = transport.accept(timeout=1)
+                conn: Optional[paramiko.channel.Channel] = transport.accept(
+                    timeout=1)
+                assert conn is not None
                 e = transport.get_exception()
                 if e:
                     raise e