diff options
Diffstat (limited to 'src/python/gudhi/tensorflow/lower_star_simplex_tree_layer.py')
-rw-r--r-- | src/python/gudhi/tensorflow/lower_star_simplex_tree_layer.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/src/python/gudhi/tensorflow/lower_star_simplex_tree_layer.py b/src/python/gudhi/tensorflow/lower_star_simplex_tree_layer.py index e1627944..8da1f7fe 100644 --- a/src/python/gudhi/tensorflow/lower_star_simplex_tree_layer.py +++ b/src/python/gudhi/tensorflow/lower_star_simplex_tree_layer.py @@ -12,8 +12,7 @@ def _LowerStarSimplexTree(simplextree, filtration, dimensions): # filtration (function values on the vertices of st), # dimensions (homology dimensions), - for s,_ in simplextree.get_filtration(): - simplextree.assign_filtration(s, -1e10) + simplextree.reset_filtration(-np.inf, 0) # Assign new filtration values for i in range(simplextree.num_vertices()): |