Merge pull request #399 from iteal/master

Fix type warning in run_tracker
talmolab · Sep 8, 2020 · 843abc6 · 843abc6
2 parents afa5de3 + d9612e2
commit 843abc6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sleap/nn/tracking.py b/sleap/nn/tracking.py
@@ -1059,7 +1059,7 @@ def run(self, frames: List[LabeledFrame]):
  connect_single_track_breaks(frames, self.instance_count)
 
 
-def run_tracker(frames: List[LabeledFrame], tracker: Tracker) -> List[LabeledFrame]:
+def run_tracker(frames: List[LabeledFrame], tracker: BaseTracker) -> List[LabeledFrame]:
  """Run a tracker on a set of labeled frames.
 
  Args: