remove more quotes

Janelia-Trackathon-2023 · Nov 28, 2023 · af482ab · af482ab
1 parent dc9db8a
commit af482ab
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/traccuracy/matchers/_ctc.py b/src/traccuracy/matchers/_ctc.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import TYPE_CHECKING
 
 import networkx as nx
@@ -24,7 +26,7 @@ class CTCMatcher(Matcher):
     for complete details.
     """
 
-    def _compute_mapping(self, gt_graph: "TrackingGraph", pred_graph: "TrackingGraph"):
+    def _compute_mapping(self, gt_graph: TrackingGraph, pred_graph: TrackingGraph):
         """Run ctc matching
 
         Args:

diff --git a/src/traccuracy/matchers/_iou.py b/src/traccuracy/matchers/_iou.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import numpy as np
 from tqdm import tqdm
 
@@ -110,7 +112,7 @@ class IOUMatcher(Matcher):
     def __init__(self, iou_threshold=0.6):
         self.iou_threshold = iou_threshold
 
-    def _compute_mapping(self, gt_graph: "TrackingGraph", pred_graph: "TrackingGraph"):
+    def _compute_mapping(self, gt_graph: TrackingGraph, pred_graph: TrackingGraph):
         """Computes IOU mapping for a set of grpahs
 
         Args: