fixup typing in network_plotting

OpenFreeEnergy · Oct 10, 2023 · a43daec · a43daec
1 parent f136b92
commit a43daec
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/openfe/utils/network_plotting.py b/openfe/utils/network_plotting.py
@@ -238,7 +238,7 @@ class EventHandler:
     selected : Optional[Union[Node, Edge]]
         Object selected by a mouse click (after mouse is up), or None if no
         object has been selected in the graph.
-    click_location : tuple[Optional[float], Optional[float]]
+    click_location : Optional[tuple[Optional[float], Optional[float]]]
         Cached location of the mousedown event, or None if mouse is up
     connections : List[int]
         list of IDs for connections to matplotlib canvas
@@ -247,7 +247,7 @@ def __init__(self, graph: GraphDrawing):
         self.graph = graph
         self.active: Optional[Union[Node, Edge]] = None
         self.selected: Optional[Union[Node, Edge]] = None
-        self.click_location: tuple[Optional[float], Optional[float]] = None, None
+        self.click_location: Optional[tuple[Optional[float], Optional[float]]] = None
         self.connections: list[int] = []
 
     def connect(self, canvas: MPL_FigureCanvasBase):
@@ -319,7 +319,7 @@ def on_mouseup(self, event: MPL_MouseEvent):
             self.active.on_mouseup(event, self.graph)
 
         self.active = None
-        self.click_location = None, None
+        self.click_location = None
         self.graph.draw()