diff options
-rw-r--r-- | src/python/gudhi/clustering/tomato.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/python/gudhi/clustering/tomato.py b/src/python/gudhi/clustering/tomato.py index 88a1a34d..1a2887bc 100644 --- a/src/python/gudhi/clustering/tomato.py +++ b/src/python/gudhi/clustering/tomato.py @@ -109,7 +109,7 @@ class Tomato: if input_type == "distance_matrix" and self.graph_type_ == "radius": X = numpy.array(X) r = self.params_["r"] - self.neighbors_ = [numpy.nonzero(l <= r) for l in X] + self.neighbors_ = [numpy.flatnonzero(l <= r) for l in X] if input_type == "distance_matrix" and self.graph_type_ == "knn": k = self.params_["k"] |