Change to get_edges_with_flag where appropriate

Janelia-Trackathon-2023 · Sep 8, 2023 · 65f4552 · 65f4552
1 parent ee98eeb
commit 65f4552
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/src/traccuracy/metrics/_ctc.py b/src/traccuracy/metrics/_ctc.py
@@ -50,19 +50,15 @@ def compute(self):
         }
         edge_error_counts = {
             "ws": len(
-                self.data.pred_graph.get_edges_with_attribute(
-                    EdgeAttr.WRONG_SEMANTIC, lambda x: x
+                self.data.pred_graph.get_edges_with_flag(
+                    EdgeAttr.WRONG_SEMANTIC
                 )
             ),
             "fp": len(
-                self.data.pred_graph.get_edges_with_attribute(
-                    EdgeAttr.FALSE_POS, lambda x: x
-                )
+                self.data.pred_graph.get_edges_with_flag(EdgeAttr.FALSE_POS)
             ),
             "fn": len(
-                self.data.gt_graph.get_edges_with_attribute(
-                    EdgeAttr.FALSE_NEG, lambda x: x
-                )
+                self.data.gt_graph.get_edges_with_flag(EdgeAttr.FALSE_NEG)
             ),
         }
         error_sum = get_weighted_error_sum(