Make helper function hidden

scikit-tda · Jan 28, 2021 · 320121a · 320121a
1 parent 76cda6f
commit 320121a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/cechmate/filtrations/alpha.py b/cechmate/filtrations/alpha.py
@@ -114,7 +114,7 @@ def alpha_build(X, filtration_current, filtration_upper):
         for sigma in filtration_current:
             filtration_current[sigma] = \
                 _squared_circumradius(X[np.asarray(sigma)])
-            for x in drop_elements(sigma):
+            for x in _drop_elements(sigma):
                 tau = x[1]
                 vertex = x[0]
                 if tau in filtration_lower and \
@@ -141,7 +141,7 @@ def alpha_build(X, filtration_current, filtration_upper):
             if np.isnan(filtration_current[sigma]):
                 filtration_current[sigma] = \
                     _squared_circumradius(X[np.asarray(sigma)])
-            for x in drop_elements(sigma):
+            for x in _drop_elements(sigma):
                 tau = x[1]
                 vertex = x[0]
                 if tau in filtration_lower and \
@@ -157,7 +157,7 @@ def alpha_build(X, filtration_current, filtration_upper):
 
     # Correct artifacts
     for omega in filtration_upper:
-        for x in drop_elements(omega):
+        for x in _drop_elements(omega):
             sigma = x[1]
             filtration_current[sigma] = min(filtration_current[sigma],
                                             filtration_upper[omega])
@@ -262,7 +262,7 @@ def _pdist(A):
 
 
 @njit
-def drop_elements(tup: tuple):
+def _drop_elements(tup: tuple):
     for x in range(len(tup)):
         empty = tup[:-1]  # Not empty, but the right size and will be mutated
         idx = 0