From 4340aaf53e5f40238ec31d6fe03cad8c853cf1a9 Mon Sep 17 00:00:00 2001 From: Jan Funke Date: Thu, 25 Apr 2024 13:48:32 -0400 Subject: [PATCH] Make black happy --- src/motile_toolbox/candidate_graph/utils.py | 12 ++---------- 1 file changed, 2 insertions(+), 10 deletions(-) diff --git a/src/motile_toolbox/candidate_graph/utils.py b/src/motile_toolbox/candidate_graph/utils.py index 93cd6e4..61307ba 100644 --- a/src/motile_toolbox/candidate_graph/utils.py +++ b/src/motile_toolbox/candidate_graph/utils.py @@ -100,19 +100,11 @@ def _compute_node_frame_dict(cand_graph: nx.DiGraph) -> dict[int, list[Any]]: return node_frame_dict -def create_kdtree( - cand_graph: nx.DiGraph, - node_ids: Iterable[Any] -) -> KDTree: - - positions = [ - cand_graph.nodes[node][NodeAttr.POS.value] - for node in node_ids - ] +def create_kdtree(cand_graph: nx.DiGraph, node_ids: Iterable[Any]) -> KDTree: + positions = [cand_graph.nodes[node][NodeAttr.POS.value] for node in node_ids] return KDTree(positions) - def add_cand_edges( cand_graph: nx.DiGraph, max_edge_distance: float,