From 2db959d99f1738af560cf9f61a9b0bc54f592e45 Mon Sep 17 00:00:00 2001 From: Elizabeth Berrigan Date: Mon, 7 Oct 2024 20:54:07 -0700 Subject: [PATCH] black --- sleap/nn/tracking.py | 18 ++++++++++++------ 1 file changed, 12 insertions(+), 6 deletions(-) diff --git a/sleap/nn/tracking.py b/sleap/nn/tracking.py index 0feff4ef2..83f994627 100644 --- a/sleap/nn/tracking.py +++ b/sleap/nn/tracking.py @@ -574,7 +574,7 @@ class Tracker(BaseTracker): max_tracking: bool = False # To enable maximum tracking. cleaner: Optional[Callable] = None # TODO: deprecate - target_instance_count: int = 0 # TODO: deprecate + target_instance_count: int = 0 # TODO: deprecate pre_cull_function: Optional[Callable] = None post_connect_single_breaks: bool = False robust_best_instance: float = 1.0 @@ -824,7 +824,9 @@ def final_pass(self, frames: List[LabeledFrame]): # "tracking." # ) self.cleaner.run(frames) - elif (self.target_instance_count or self.max_tracks) and self.post_connect_single_breaks: + elif ( + self.target_instance_count or self.max_tracks + ) and self.post_connect_single_breaks: if not self.target_instance_count: # If target_instance_count is not set, use max_tracks instead # target_instance_count not available in the GUI @@ -855,7 +857,7 @@ def make_tracker_by_name( of_max_levels: int = 3, save_shifted_instances: bool = False, # Pre-tracking options to cull instances - target_instance_count: int = 0, # TODO: deprecate target_instance_count + target_instance_count: int = 0, # TODO: deprecate target_instance_count pre_cull_to_target: bool = False, pre_cull_iou_threshold: Optional[float] = None, # Post-tracking options to connect broken tracks @@ -946,7 +948,7 @@ def pre_cull_function(inst_list): pre_cull_function=pre_cull_function, max_tracking=max_tracking, max_tracks=max_tracks, - target_instance_count=target_instance_count, # TODO: deprecate target_instance_count + target_instance_count=target_instance_count, # TODO: deprecate target_instance_count post_connect_single_breaks=post_connect_single_breaks, ) @@ -967,7 +969,9 @@ def pre_cull_function(inst_list): return kalman_obj elif kf_init_frame_count and not (max_tracks or target_instance_count): - raise ValueError("Kalman filter requires max tracks or target instance count.") + raise ValueError( + "Kalman filter requires max tracks or target instance count." + ) else: return tracker_obj @@ -1381,7 +1385,9 @@ def cull_function(inst_list): if init_tracker.pre_cull_function is None: init_tracker.pre_cull_function = cull_function - print(f"Using {init_tracker.get_name()} to track {init_frame_count} frames for Kalman filters.") + print( + f"Using {init_tracker.get_name()} to track {init_frame_count} frames for Kalman filters." + ) return cls( init_tracker=init_tracker,