Make black happy

funkelab · Apr 25, 2024 · 4340aaf · 4340aaf
1 parent ed7930b
commit 4340aaf
Showing 1 changed file with 2 additions and 10 deletions.
diff --git 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,