Merge pull request #94 from lmanan/main

funkelab · Apr 24, 2024 · 1216dec · 1216dec
2 parents 5415cd4 + 06608fe
commit 1216dec
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/motile/solver.py b/motile/solver.py
@@ -329,11 +329,11 @@ def get_selected_subgraph(
         selected_graph = TrackGraph()
 
         for node_id, node in self.graph.nodes.items():
-            if solution[node_selected[node_id]]:
+            if solution[node_selected[node_id]] > 0.5:
                 selected_graph.add_node(node_id, node)
 
         for edge_id, edge in self.graph.edges.items():
-            if solution[edge_selected[edge_id]]:
+            if solution[edge_selected[edge_id]] > 0.5:
                 selected_graph.add_edge(edge_id, edge)
 
         return selected_graph