diff options
| author | Marc Glisse <marc.glisse@inria.fr> | 2020-04-20 10:38:41 +0200 |
|---|---|---|
| committer | Marc Glisse <marc.glisse@inria.fr> | 2020-04-20 10:38:41 +0200 |
| commit | 1c1a99074049e4ff04fa28e7d6e1b6fc2067397a (patch) | |
| tree | 9cad7a7883ad9a8e7bbdde8f13443379a4cc39a5 /src/python/gudhi/point_cloud | |
| parent | 8d9611206603f4f7506fe77a0273c73c9d67716b (diff) | |
Add __license__
Diffstat (limited to 'src/python/gudhi/point_cloud')
| -rw-r--r-- | src/python/gudhi/point_cloud/dtm.py | 4 | ||||
| -rw-r--r-- | src/python/gudhi/point_cloud/knn.py | 8 |
2 files changed, 11 insertions, 1 deletions
diff --git a/src/python/gudhi/point_cloud/dtm.py b/src/python/gudhi/point_cloud/dtm.py index 58dec536..13e16d24 100644 --- a/src/python/gudhi/point_cloud/dtm.py +++ b/src/python/gudhi/point_cloud/dtm.py @@ -9,6 +9,10 @@ from .knn import KNearestNeighbors +__author__ = "Marc Glisse" +__copyright__ = "Copyright (C) 2020 Inria" +__license__ = "MIT" + class DistanceToMeasure: """ diff --git a/src/python/gudhi/point_cloud/knn.py b/src/python/gudhi/point_cloud/knn.py index d7cf0b2a..4017e498 100644 --- a/src/python/gudhi/point_cloud/knn.py +++ b/src/python/gudhi/point_cloud/knn.py @@ -11,6 +11,10 @@ import numpy # TODO: https://github.com/facebookresearch/faiss +__author__ = "Marc Glisse" +__copyright__ = "Copyright (C) 2020 Inria" +__license__ = "MIT" + class KNearestNeighbors: """ @@ -156,7 +160,9 @@ class KNearestNeighbors: assert self.metric == "minkowski" p = self.params["p"] Y = ep.astensor(self.ref_points) - neighbor_pts = Y[neighbors,] + neighbor_pts = Y[ + neighbors, + ] diff = neighbor_pts - X[:, None, :] if isinstance(diff, ep.PyTorchTensor): # https://github.com/jonasrauber/eagerpy/issues/6 |