UltrackWidget: include z axis on 3d tracks (#164)

pyproject.toml

@@ -7,7 +7,7 @@ name = "ultrack"
 dynamic = ["version"]
 description = "Large-scale multi-hypotheses cell tracking"
 readme = "README.md"
-license = ""
+license = "BSD-3-Clause"
 requires-python = ">=3.9,<3.13"
 authors = [
     { name = "Jordao Bragantini", email = "[email protected]" },

ultrack/widgets/ultrackwidget/workflows.py

@@ -257,8 +257,11 @@ def run(
                     scale=detection.scale,
                 )
 
+                is_3d = segments.ndim == 4
+                dims = ["z", "y", "x"] if is_3d else ["y", "x"]
+
                 yield Tracks(
-                    data=tracks_df[["track_id", "t", "y", "x"]].values,
+                    data=tracks_df[["track_id", "t", *dims]].values,
                     graph=graph,
                     name="Tracks",
                     scale=detection.scale[1:],