Skip to content

Commit d92d956

Browse files
Raise meaningful error message for missing trackastra package (#1184)
* Raise meaningful error message for missing trackastra package * Update micro_sam/multi_dimensional_segmentation.py --------- Co-authored-by: Constantin Pape <constantin.pape@informatik.uni-goettingen.de>
1 parent ca2e965 commit d92d956

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

micro_sam/multi_dimensional_segmentation.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,8 @@
3030
from trackastra.tracking import graph_to_ctc, graph_to_napari_tracks
3131
except ImportError:
3232
Trackastra = None
33+
graph_to_ctc = None
34+
graph_to_napari_tracks = None
3335

3436

3537
from . import util
@@ -621,6 +623,11 @@ def track_across_frames(
621623
with each dict encoding a lineage, where keys correspond to parent track ids.
622624
Each key either maps to a list with two child track ids (cell division) or to an empty list (no division).
623625
"""
626+
if Trackastra is None:
627+
raise RuntimeError(
628+
"The automatic tracking functionality requires trackastra. You can install it via 'pip install trackastra'."
629+
)
630+
624631
_, pbar_init, pbar_update, pbar_close = util.handle_pbar(verbose, pbar_init=pbar_init, pbar_update=pbar_update)
625632

626633
if gap_closing is not None and gap_closing > 0:

0 commit comments

Comments
 (0)