Skip to content

Commit

Permalink
Missing status update to Tracker.track
Browse files Browse the repository at this point in the history
  • Loading branch information
JoOkuma authored Mar 21, 2024
1 parent 6f55b26 commit 5b9c72d
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions ultrack/core/tracker.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,6 +91,7 @@ def solve(self, *args, **kwargs) -> None:
@functools.wraps(track)
def track(self, *args, **kwargs) -> None:
track(config=self.config, *args, **kwargs)
self.status = TrackerStatus.SOLVED

def _assert_solved(self) -> None:
"""Raise an error if the tracking is not solved."""
Expand All @@ -117,14 +118,13 @@ def to_pandas(self, *args, **kwargs) -> pd.DataFrame:
return tracks_df

@functools.wraps(tracks_to_zarr)
def to_zarr(
self, *, tracks_df: Optional[pd.DataFrame] = None, **kwargs
) -> zarr.Array:
def to_zarr(self, **kwargs) -> zarr.Array:
self._assert_solved()
tracks_df = kwargs.pop("tracks_df")
if tracks_df is None:
tracks_df, _ = to_tracks_layer(self.config)
z = tracks_to_zarr(self.config, tracks_df=tracks_df, **kwargs)
return z
segments = tracks_to_zarr(self.config, tracks_df=tracks_df, **kwargs)
return segments

@functools.wraps(tracks_layer_to_trackmate)
def to_trackmate(self, tracks_df: Optional[pd.DataFrame] = None) -> str:
Expand Down

0 comments on commit 5b9c72d

Please sign in to comment.