Import Matched in matchers init and change imports accordingly

src/traccuracy/matchers/__init__.py

@@ -25,8 +25,9 @@
 While we specify ground truth and prediction, it is possible to
 write a matching function that matches two arbitrary tracking solutions.
 """
+from ._base import Matched
 from ._compute_overlap import get_labels_with_overlap
 from ._ctc import CTCMatcher
 from ._iou import IOUMatcher
 
-__all__ = ["CTCMatcher", "IOUMatcher", "get_labels_with_overlap"]
+__all__ = ["CTCMatcher", "IOUMatcher", "get_labels_with_overlap", "Matched"]

src/traccuracy/metrics/_base.py

@@ -2,7 +2,7 @@
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from traccuracy.matchers._base import Matched
+    from traccuracy.matchers import Matched
 
 
 class Metric(ABC):

src/traccuracy/metrics/_ctc.py

@@ -6,7 +6,7 @@
 from ._base import Metric
 
 if TYPE_CHECKING:
-    from ._base import Matched
+    from traccuracy.matchers import Matched
 
 
 class AOGMMetrics(Metric):

src/traccuracy/metrics/_track_overlap.py

@@ -19,7 +19,7 @@
 from ._base import Metric
 
 if TYPE_CHECKING:
-    from ._base import Matched
+    from traccuracy.matchers import Matched
 
 
 def _mapping_to_dict(mapping: List[Tuple[Any, Any]]) -> Dict[Any, List[Any]]:

src/traccuracy/track_errors/_ctc.py

@@ -7,7 +7,7 @@
 from traccuracy import EdgeAttr, NodeAttr
 
 if TYPE_CHECKING:
-    from traccuracy.matchers._base import Matched
+    from traccuracy.matchers import Matched
 
 logger = logging.getLogger(__name__)
 

src/traccuracy/track_errors/divisions.py

@@ -8,7 +8,7 @@
 from traccuracy._utils import find_gt_node_matches, find_pred_node_matches
 
 if TYPE_CHECKING:
-    from traccuracy.matchers._base import Matched
+    from traccuracy.matchers import Matched
 
 logger = logging.getLogger(__name__)
 

tests/metrics/test_divisions.py

@@ -1,5 +1,5 @@
 from traccuracy import TrackingGraph
-from traccuracy.matchers._base import Matched
+from traccuracy.matchers import Matched
 from traccuracy.metrics._divisions import DivisionMetrics
 
 from tests.test_utils import get_division_graphs

tests/metrics/test_track_overlap_metrics.py

@@ -3,7 +3,7 @@
 import networkx as nx
 import pytest
 from traccuracy import TrackingGraph
-from traccuracy.matchers._base import Matched
+from traccuracy.matchers import Matched
 from traccuracy.metrics._track_overlap import TrackOverlapMetrics, _mapping_to_dict
 
 

tests/track_errors/test_ctc_errors.py

@@ -1,7 +1,7 @@
 import networkx as nx
 import numpy as np
 from traccuracy._tracking_graph import EdgeAttr, NodeAttr, TrackingGraph
-from traccuracy.matchers._base import Matched
+from traccuracy.matchers import Matched
 from traccuracy.track_errors._ctc import get_edge_errors, get_vertex_errors
 
 

tests/track_errors/test_divisions.py

@@ -2,7 +2,7 @@
 import numpy as np
 import pytest
 from traccuracy import NodeAttr, TrackingGraph
-from traccuracy.matchers._base import Matched
+from traccuracy.matchers import Matched
 from traccuracy.track_errors.divisions import (
     _classify_divisions,
     _correct_shifted_divisions,