fix inf points in atol fitting

GUDHI · Sep 18, 2023 · 3dab4f8 · 3dab4f8
1 parent 9978b84
commit 3dab4f8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/python/gudhi/representations/vector_methods.py b/src/python/gudhi/representations/vector_methods.py
@@ -781,6 +781,10 @@ def fit(self, X, y=None, sample_weight=None):
         """
         if not hasattr(self.quantiser, 'fit'):
             raise TypeError("quantiser %s has no `fit` attribute." % (self.quantiser))
+
+        # In fitting we remove infinite death time points so that every center is finite
+        X = [dgm[~np.isinf(dgm).any(axis=1), :] for dgm in X]
+
         if sample_weight is None:
             sample_weight = [self.get_weighting_method()(measure) for measure in X]