From 923127cdad367c4cf421dabecacc27dc2e3f97be Mon Sep 17 00:00:00 2001 From: dummyindex Date: Mon, 22 Jul 2024 03:59:27 -0400 Subject: [PATCH] feat: Refactor find_maps function in sc_mapping.py --- livecellx/core/sc_mapping.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/livecellx/core/sc_mapping.py b/livecellx/core/sc_mapping.py index 82c7998..a2ad0b5 100644 --- a/livecellx/core/sc_mapping.py +++ b/livecellx/core/sc_mapping.py @@ -58,16 +58,21 @@ def find_maps( scs_from: List[SingleCellStatic], scs_to: List[SingleCellStatic], metric_threshold=0.3, - metric_key="iomin", + metric_key=None, min_map_num=None, metric="iomin", ): if metric == "iou": + if metric_key is None: + metric_key = "iou" compute_scs_iou(scs_from, scs_to, metric_key) elif metric == "iomin": + if metric_key is None: + metric_key = "iomin" compute_scs_iomin(scs_from, scs_to, metric_key) else: raise ValueError(f"Unknown metric: {metric}") + scs_map = {} for sc1 in scs_from: scs_map[sc1] = []