handle empty clusters

Kapoorlabs-CAPED · Nov 3, 2023 · d3b1ddd · d3b1ddd
1 parent 5aa98e7
commit d3b1ddd
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/src/napatrackmater/Trackvector.py b/src/napatrackmater/Trackvector.py
@@ -907,6 +907,15 @@ def unsupervised_clustering(
             track_arrays_array.index(track_arrays)
         ]
         shape_dynamic_cosine_distance = pdist(clusterable_track_array, metric=metric)
+        if (
+            np.isnan(shape_dynamic_cosine_distance).any()
+            or np.isinf(shape_dynamic_cosine_distance).any()
+        ):
+            print(
+                "Cosine distance matrix contains NaN or infinite values. Returning an empty linkage matrix."
+            )
+            return
+
         shape_dynamic_linkage_matrix = linkage(
             shape_dynamic_cosine_distance, method=method
         )

diff --git a/src/napatrackmater/_version.py b/src/napatrackmater/_version.py
@@ -1,2 +1,2 @@
-__version__ = version = "4.4.1"
-__version_tuple__ = version_tuple = (4, 4, 1)
+__version__ = version = "4.4.2"
+__version_tuple__ = version_tuple = (4, 4, 2)