From f74c71ca8e474ff927cae029ea63329d30293582 Mon Sep 17 00:00:00 2001
From: Marc Glisse <marc.glisse@inria.fr>
Date: Fri, 27 Mar 2020 13:43:58 +0100
Subject: Improve coverage

---
 src/python/gudhi/point_cloud/dtm.py | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'src/python/gudhi/point_cloud/dtm.py')

diff --git a/src/python/gudhi/point_cloud/dtm.py b/src/python/gudhi/point_cloud/dtm.py
index 839e7452..541b74a6 100644
--- a/src/python/gudhi/point_cloud/dtm.py
+++ b/src/python/gudhi/point_cloud/dtm.py
@@ -30,6 +30,8 @@ class DTM:
             X (numpy.array): coordinates for mass points
         """
         if self.params.setdefault("metric", "euclidean") != "neighbors":
+            # KNN gives sorted distances, which is unnecessary here.
+            # Maybe add a parameter to say we don't need sorting?
             self.knn = KNN(self.k, return_index=False, return_distance=True, **self.params)
             self.knn.fit(X)
         return self
-- 
cgit v1.2.3