diff options
| author | Hind-M <hind.montassif@gmail.com> | 2021-08-30 15:34:15 +0200 |
|---|---|---|
| committer | Hind-M <hind.montassif@gmail.com> | 2021-08-30 15:34:15 +0200 |
| commit | 7ea4e020af2fa8bf2fdfefe85ca24a1bcc2d08e1 (patch) | |
| tree | bf06d09a4746eeec436e5f1a1eb703bce617c76c | |
| parent | 2024c0af61c1b14e50eccfae9a0011cb061b16d2 (diff) | |
Fix dtm and knn failing tests
| -rw-r--r-- | src/python/gudhi/point_cloud/knn.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/python/gudhi/point_cloud/knn.py b/src/python/gudhi/point_cloud/knn.py index 7a5616e3..dec5f88f 100644 --- a/src/python/gudhi/point_cloud/knn.py +++ b/src/python/gudhi/point_cloud/knn.py @@ -296,10 +296,10 @@ class KNearestNeighbors: if self.return_index: if self.return_distance: distances, neighbors = mat.Kmin_argKmin(k, dim=1) - if distances.isnan().any(): + if torch.isnan(distances).any(): import warnings warnings.warn("NaN value encountered while computing 'distances'", RuntimeWarning) - if distances.isinf().any(): + if torch.isinf(distances).any(): import warnings warnings.warn("Overflow encountered while computing 'distances'", RuntimeWarning) if p != numpy.inf: @@ -310,10 +310,10 @@ class KNearestNeighbors: return neighbors if self.return_distance: distances = mat.Kmin(k, dim=1) - if distances.isnan().any(): + if torch.isnan(distances).any(): import warnings warnings.warn("NaN value encountered while computing 'distances'", RuntimeWarning) - if distances.isinf().any(): + if torch.isinf(distances).any(): import warnings warnings.warn("Overflow encountered while computing 'distances'", RuntimeWarning) if p != numpy.inf: |