diff --git a/sleap/nn/tracking.py b/sleap/nn/tracking.py index d14169119..dc2f0a367 100644 --- a/sleap/nn/tracking.py +++ b/sleap/nn/tracking.py @@ -653,10 +653,7 @@ def infer_next_timestep(self, t: Optional[int] = None) -> int: self.track_matching_queue_dict, key=lambda track: len(self.track_matching_queue_dict[track]), ) - return ( - self.track_matching_queue_dict[track_with_max_instances][-1].t - + 1 - ) + return 1 + self.track_matching_queue_dict[track_with_max_instances][-1].t # Default to last timestep + 1 if available. if not self.has_max_tracking and len(self.track_matching_queue) > 0: @@ -1521,7 +1518,9 @@ def run_tracker(frames: List[LabeledFrame], tracker: BaseTracker) -> List[Labele # Run tracking on every frame for lf in frames: # Use only the predicted instances - instances = [inst for inst in lf.instances if isinstance(inst, PredictedInstance)] + instances = [ + inst for inst in lf.instances if isinstance(inst, PredictedInstance) + ] # Clear the tracks for inst in instances: