From e7010b343079bfddee7740b9859121b3b33968ab Mon Sep 17 00:00:00 2001 From: Jordao Bragantini Date: Thu, 5 Feb 2026 11:14:39 -0800 Subject: [PATCH 1/3] Fixing loading of matching weights --- src/tracksdata/metrics/_ctc_metrics.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/tracksdata/metrics/_ctc_metrics.py b/src/tracksdata/metrics/_ctc_metrics.py index 39640508..0f5e27db 100644 --- a/src/tracksdata/metrics/_ctc_metrics.py +++ b/src/tracksdata/metrics/_ctc_metrics.py @@ -104,7 +104,7 @@ def _match_single_frame( # loading original group ids and filtering by the matches _mapped_ref = ref_group[reference_graph_key][rows_id].to_list() _mapped_comp = comp_group[input_graph_key][cols_id].to_list() - _ious = weights[rows_id, cols_id].tolist() + _ious = np.asarray(weights[rows_id, cols_id]).ravel().tolist() LOG.info("Done!") From c1cfffdfabaf63b20d8e7baa2c8197cd162246e0 Mon Sep 17 00:00:00 2001 From: Jordao Bragantini Date: Thu, 5 Feb 2026 11:19:10 -0800 Subject: [PATCH 2/3] fixing testing --- src/tracksdata/graph/_test/test_graph_backends.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/tracksdata/graph/_test/test_graph_backends.py b/src/tracksdata/graph/_test/test_graph_backends.py index fc52fe59..86c9693f 100644 --- a/src/tracksdata/graph/_test/test_graph_backends.py +++ b/src/tracksdata/graph/_test/test_graph_backends.py @@ -2288,7 +2288,7 @@ def _fill_mock_geff_graph(graph_backend: BaseGraph) -> None: ) graph_backend.add_node_attr_key(DEFAULT_ATTR_KEYS.MASK, pl.Object) graph_backend.add_node_attr_key(DEFAULT_ATTR_KEYS.TRACKLET_ID, pl.Int64) - graph_backend.add_node_attr_key("ndfeature", pl.Object) + graph_backend.add_node_attr_key("ndfeature", pl.Array(pl.Float64, (3, 1)), np.ones((3, 1))) graph_backend.add_edge_attr_key("weight", pl.Float64) From 06400db3f000fa25ca542c3359ca567a66a21d37 Mon Sep 17 00:00:00 2001 From: Jordao Bragantini Date: Thu, 5 Feb 2026 13:13:53 -0800 Subject: [PATCH 3/3] addressing case when _ious is empty --- src/tracksdata/metrics/_ctc_metrics.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/tracksdata/metrics/_ctc_metrics.py b/src/tracksdata/metrics/_ctc_metrics.py index 0f5e27db..876cb54a 100644 --- a/src/tracksdata/metrics/_ctc_metrics.py +++ b/src/tracksdata/metrics/_ctc_metrics.py @@ -104,7 +104,8 @@ def _match_single_frame( # loading original group ids and filtering by the matches _mapped_ref = ref_group[reference_graph_key][rows_id].to_list() _mapped_comp = comp_group[input_graph_key][cols_id].to_list() - _ious = np.asarray(weights[rows_id, cols_id]).ravel().tolist() + _ious = weights[rows_id, cols_id] + _ious = _ious.tolist() if _ious.size > 0 else [] LOG.info("Done!")