✨ Fix some linting issues

Bycelium · Feb 11, 2022 · dbddeb2 · dbddeb2
1 parent 3842496
commit dbddeb2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/pyflow/blocks/codeblock.py b/pyflow/blocks/codeblock.py
@@ -3,7 +3,7 @@
 
 """ Module for the base Code Block."""
 
-from typing import OrderedDict, Tuple
+from typing import TYPE_CHECKING, OrderedDict, Tuple
 
 from ansi2html import Ansi2HTMLConverter
 from PyQt5.QtGui import QColor, QPen
@@ -16,6 +16,9 @@
 
 ansi2html_converter = Ansi2HTMLConverter()
 
+if TYPE_CHECKING:
+    from PyQt5.QtWidgets import QGraphicsSceneHoverEvent
+
 
 class CodeBlock(ExecutableBlock):
 

diff --git a/pyflow/graphics/view.py b/pyflow/graphics/view.py
@@ -491,7 +491,7 @@ def drag_edge(self, event: QMouseEvent, action="press"):
                 LOGGER.debug("Start draging edge from existing socket.")
                 return
             # If it is the add edge button, create a new socket and a new edge from it.
-            elif (
+            if (
                 isinstance(item_at_click, AddEdgeButton)
                 and self.mode != self.MODE_EDGE_DRAG
             ):