From 225a5316375ed5e7e64f5b8b007b6f11e8419668 Mon Sep 17 00:00:00 2001
From: Caroline Malin-Mayor <malinmayorc@janelia.hhmi.org>
Date: Wed, 21 Aug 2024 13:41:46 -0400
Subject: [PATCH] Add back in bonus task

---
 solution.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/solution.py b/solution.py
index 29803ef..70e6024 100644
--- a/solution.py
+++ b/solution.py
@@ -876,6 +876,16 @@ def add_gt_annotations(gt_tracks, cand_graph, segmentation):
 # <p>Now, similar to before, we make the solver by adding costs and constraints. You can copy your best set of costs and constraints from before. It does not matter what weights and constants you choose. However, this time we just return the solver, rather than actually solving.</p>
 # </div>
 
+# %% tags=["task"]
+def get_ssvm_solver(cand_graph):
+
+    cand_trackgraph = motile.TrackGraph(cand_graph, frame_attribute="t")
+    solver = motile.Solver(cand_trackgraph)
+
+    ### YOUR CODE HERE ###
+    return solver
+
+
 # %%
 def get_ssvm_solver(cand_graph):