changed dimensions into homology_dimensions

3 files changed, 9 insertions, 9 deletions

diff --git a/src/python/gudhi/tensorflow/cubical_layer.py b/src/python/gudhi/tensorflow/cubical_layer.py
index d68c7556..3304e719 100644
--- a/src/python/gudhi/tensorflow/cubical_layer.py
+++ b/src/python/gudhi/tensorflow/cubical_layer.py
@@ -37,17 +37,17 @@ class CubicalLayer(tf.keras.layers.Layer):
     """
     TensorFlow layer for computing the persistent homology of a cubical complex
     """
-    def __init__(self, dimensions, min_persistence=None, homology_coeff_field=11, **kwargs):
+    def __init__(self, homology_dimensions, min_persistence=None, homology_coeff_field=11, **kwargs):
         """
         Constructor for the CubicalLayer class
 
         Parameters:
-            dimensions (List[int]): homology dimensions
+            homology_dimensions (List[int]): list of homology dimensions
             min_persistence (List[float]): minimum distance-to-diagonal of the points in the output persistence diagrams (default None, in which case 0. is used for all dimensions)
             homology_coeff_field (int): homology field coefficient. Must be a prime number. Default value is 11. Max is 46337.
         """
         super().__init__(dynamic=True, **kwargs)
-        self.dimensions = dimensions
+        self.dimensions = homology_dimensions
         self.min_persistence = min_persistence if min_persistence != None else [0.] * len(self.dimensions)
         self.hcf = homology_coeff_field
         assert len(self.min_persistence) == len(self.dimensions)
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 4ec3f7c7..5a8e5b75 100644
--- a/src/python/gudhi/tensorflow/lower_star_simplex_tree_layer.py
+++ b/src/python/gudhi/tensorflow/lower_star_simplex_tree_layer.py
@@ -43,18 +43,18 @@ class LowerStarSimplexTreeLayer(tf.keras.layers.Layer):
     """
     TensorFlow layer for computing lower-star persistence out of a simplex tree
     """
-    def __init__(self, simplextree, dimensions, min_persistence=None, homology_coeff_field=11, **kwargs):
+    def __init__(self, simplextree, homology_dimensions, min_persistence=None, homology_coeff_field=11, **kwargs):
         """
         Constructor for the LowerStarSimplexTreeLayer class
   
         Parameters:
             simplextree (gudhi.SimplexTree): underlying simplex tree. Its vertices MUST be named with integers from 0 to n-1, where n is its number of vertices. Note that its filtration values are modified in each call of the class.
-            dimensions (List[int]): homology dimensions
+            homology_dimensions (List[int]): list of homology dimensions
             min_persistence (List[float]): minimum distance-to-diagonal of the points in the output persistence diagrams (default None, in which case 0. is used for all dimensions)
             homology_coeff_field (int): homology field coefficient. Must be a prime number. Default value is 11. Max is 46337.
         """
         super().__init__(dynamic=True, **kwargs)
-        self.dimensions  = dimensions
+        self.dimensions  = homology_dimensions
         self.simplextree = simplextree
         self.min_persistence = min_persistence if min_persistence != None else [0. for _ in range(len(self.dimensions))]
         self.hcf = homology_coeff_field
diff --git a/src/python/gudhi/tensorflow/rips_layer.py b/src/python/gudhi/tensorflow/rips_layer.py
index fca336f3..2a73472c 100644
--- a/src/python/gudhi/tensorflow/rips_layer.py
+++ b/src/python/gudhi/tensorflow/rips_layer.py
@@ -40,19 +40,19 @@ class RipsLayer(tf.keras.layers.Layer):
     """
     TensorFlow layer for computing Rips persistence out of a point cloud
     """
-    def __init__(self, dimensions, maximum_edge_length=np.inf, min_persistence=None, homology_coeff_field=11, **kwargs):
+    def __init__(self, homology_dimensions, maximum_edge_length=np.inf, min_persistence=None, homology_coeff_field=11, **kwargs):
         """
         Constructor for the RipsLayer class
 
         Parameters:
             maximum_edge_length (float): maximum edge length for the Rips complex 
-            dimensions (List[int]): homology dimensions
+            homology_dimensions (List[int]): list of homology dimensions
             min_persistence (List[float]): minimum distance-to-diagonal of the points in the output persistence diagrams (default None, in which case 0. is used for all dimensions)
             homology_coeff_field (int): homology field coefficient. Must be a prime number. Default value is 11. Max is 46337.
         """
         super().__init__(dynamic=True, **kwargs)
         self.max_edge = maximum_edge_length
-        self.dimensions = dimensions
+        self.dimensions = homology_dimensions
         self.min_persistence = min_persistence if min_persistence != None else [0. for _ in range(len(self.dimensions))]
         self.hcf = homology_coeff_field
         assert len(self.min_persistence) == len(self.dimensions)