Merge pull request #144 from funkelab/bugfix-seg-page-none

Bugfix: Get segmentation from page only if it exists
funkelab · Feb 19, 2025 · d4d778a · d4d778a
2 parents 4779c40 + 65fa669
commit d4d778a
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/motile_tracker/import_export/menus/import_external_tracks_dialog.py b/src/motile_tracker/import_export/menus/import_external_tracks_dialog.py
@@ -203,13 +203,12 @@ def _finish(self) -> None:
 
         if self.menu_widget.segmentation_checkbox.isChecked():
             self.segmentation_page._load_segmentation()
+            segmentation = self.segmentation_page.segmentation
         else:
-            self.segmentation_page.segmentation = None
+            segmentation = None
 
         try:
-            self.tracks = tracks_from_df(
-                df, self.segmentation_page.segmentation, scale, features
-            )
+            self.tracks = tracks_from_df(df, segmentation, scale, features)
 
         except ValueError as e:
             QMessageBox.critical(self, "Error", f"Failed to load tracks: {e}")