From b2ffcacf37293e82ea4140beeb7a1f576e0cc658 Mon Sep 17 00:00:00 2001 From: Andrew Shum Date: Wed, 21 Aug 2024 00:26:33 -0400 Subject: [PATCH] added result check --- modules/clustering/clustering.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/modules/clustering/clustering.py b/modules/clustering/clustering.py index a26962e..fb8c7e0 100644 --- a/modules/clustering/clustering.py +++ b/modules/clustering/clustering.py @@ -73,10 +73,12 @@ def run( # if far enough, send current cluster, initialize new one if distance_from_last_point > self.max_cluster_distance or direction_switched: - new_cluster = detection_cluster.DetectionCluster.create(self.cluster) + result, new_cluster = detection_cluster.DetectionCluster.create(self.cluster) + if not result: + return False, None self.cluster = [] self.cluster.append(point) - return new_cluster + return True, new_cluster # if close enough, cluster together self.cluster.append(point)