lint

golemfactory · Mar 4, 2020 · 20e6176 · 20e6176
1 parent 848360e
commit 20e6176
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/golem/task/server/queue_.py b/golem/task/server/queue_.py
@@ -115,7 +115,9 @@ def msg_queue_connection_established(
             conn_id,
             node_id,
     ):
-        session: 'TaskSession' = protocol.session
+        session = protocol.session
+        if typing.TYPE_CHECKING:
+            assert isinstance(session, TaskSession)
         try:
             if self.sessions[node_id] is not None:
                 # There is a session already established