From 3d4ad38ef88e74cb02b584aad8dec6510c7060d4 Mon Sep 17 00:00:00 2001 From: Steven Hiscocks Date: Thu, 18 Jul 2024 16:17:33 +0100 Subject: [PATCH] Handle distance of 0 correctly in TrackToTruth associator In edge case where distance was exactly zero, min_dist would evaluate false. --- stonesoup/dataassociator/tracktotrack.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/stonesoup/dataassociator/tracktotrack.py b/stonesoup/dataassociator/tracktotrack.py index 789d35527..d8e689979 100644 --- a/stonesoup/dataassociator/tracktotrack.py +++ b/stonesoup/dataassociator/tracktotrack.py @@ -275,7 +275,7 @@ def associate_tracks(self, tracks_set: Set[Track], truth_set: Set[GroundTruthPat for track_state in Track.last_timestamp_generator(track): - min_dist = None + min_dist = self.association_threshold min_truth = None for truth in truth_set: @@ -290,11 +290,7 @@ def associate_tracks(self, tracks_set: Set[Track], truth_set: Set[GroundTruthPat continue distance = self.measure(track_state, truth_state) - if min_dist and distance < min_dist: - min_dist = distance - min_truth = truth - elif not min_dist \ - and distance < self.association_threshold: + if distance < min_dist: min_dist = distance min_truth = truth