From 0bf357f6346fab6edf96d580a9195c2acbb79bae Mon Sep 17 00:00:00 2001
From: Hind-M <hind.montassif@gmail.com>
Date: Mon, 25 Oct 2021 10:23:45 +0200
Subject: Revert "Add warnings in dtm.py for DistanceToMeasure and DTMDensity"

This reverts commit f461f050ee8bad509814b4851ab7ae8f43502962.
---
 src/python/gudhi/point_cloud/dtm.py | 11 -----------
 src/python/test/test_dtm.py         | 10 +---------
 2 files changed, 1 insertion(+), 20 deletions(-)

(limited to 'src')

diff --git a/src/python/gudhi/point_cloud/dtm.py b/src/python/gudhi/point_cloud/dtm.py
index 96a9e7bf..55ac58e6 100644
--- a/src/python/gudhi/point_cloud/dtm.py
+++ b/src/python/gudhi/point_cloud/dtm.py
@@ -9,7 +9,6 @@
 
 from .knn import KNearestNeighbors
 import numpy as np
-import warnings
 
 __author__ = "Marc Glisse"
 __copyright__ = "Copyright (C) 2020 Inria"
@@ -67,11 +66,6 @@ class DistanceToMeasure:
         distances = distances ** self.q
         dtm = distances.sum(-1) / self.k
         dtm = dtm ** (1.0 / self.q)
-        with warnings.catch_warnings():
-            import torch
-            if isinstance(dtm, torch.Tensor):
-                if not(torch.isfinite(dtm).all()):
-                    warnings.warn("Overflow/infinite value encountered while computing 'dtm'", RuntimeWarning)
         # We compute too many powers, 1/p in knn then q in dtm, 1/q in dtm then q or some log in the caller.
         # Add option to skip the final root?
         return dtm
@@ -169,11 +163,6 @@ class DTMDensity:
             distances = self.knn.transform(X)
         distances = distances ** q
         dtm = (distances * weights).sum(-1)
-        with warnings.catch_warnings():
-            import torch
-            if isinstance(dtm, torch.Tensor):
-                if not(torch.isfinite(dtm).all()):
-                    warnings.warn("Overflow/infinite value encountered while computing 'dtm' for density", RuntimeWarning)
         if self.normalize:
             dtm /= (np.arange(1, k + 1) ** (q / dim) * weights).sum()
         density = dtm ** (-dim / q)
diff --git a/src/python/test/test_dtm.py b/src/python/test/test_dtm.py
index 52468d0f..c29471cf 100755
--- a/src/python/test/test_dtm.py
+++ b/src/python/test/test_dtm.py
@@ -97,15 +97,7 @@ def test_dtm_overflow_warnings():
         for impl in impl_warn:
             dtm = DistanceToMeasure(2, q=10000, implementation=impl)
             r = dtm.fit_transform(pts)
-        assert len(w) == 3
+        assert len(w) == 2
         for i in range(len(w)):
             assert issubclass(w[i].category, RuntimeWarning)
             assert "Overflow" in str(w[i].message)
-
-def test_density_overflow_warning():
-    distances = numpy.array([[10., 100.], [10000000000000., 10.]])
-    with warnings.catch_warnings(record=True) as w:
-        density = DTMDensity(k=2, q=100000, implementation="keops", dim=1).fit_transform(distances)
-        assert len(w) == 1
-        assert issubclass(w[0].category, RuntimeWarning)
-        assert "Overflow" in str(w[0].message)
-- 
cgit v1.2.3