summaryrefslogtreecommitdiff
path: root/src/python/gudhi
diff options
context:
space:
mode:
authorMathieuCarriere <mathieu.carriere3@gmail.com>2021-11-17 00:18:33 +0100
committerMathieuCarriere <mathieu.carriere3@gmail.com>2021-11-17 00:18:33 +0100
commit7b83812e37986c9adf9cccaeab360f1d4ffa846f (patch)
tree9bb8926680b6eddf00279a2c22a9f86634552be3 /src/python/gudhi
parent7e71aff27074874acdfd3530826b3f10709e346c (diff)
fix doc
Diffstat (limited to 'src/python/gudhi')
-rw-r--r--src/python/gudhi/tensorflow/cubical_layer.py2
-rw-r--r--src/python/gudhi/tensorflow/lower_star_simplex_tree_layer.py2
-rw-r--r--src/python/gudhi/tensorflow/rips_layer.py2
3 files changed, 3 insertions, 3 deletions
diff --git a/src/python/gudhi/tensorflow/cubical_layer.py b/src/python/gudhi/tensorflow/cubical_layer.py
index 0971a446..d07a4cd8 100644
--- a/src/python/gudhi/tensorflow/cubical_layer.py
+++ b/src/python/gudhi/tensorflow/cubical_layer.py
@@ -42,7 +42,7 @@ class CubicalLayer(tf.keras.layers.Layer):
Constructor for the CubicalLayer class
Parameters:
- dimensions (list of int): homology dimensions
+ dimensions (List[int]): homology dimensions
"""
super().__init__(dynamic=True, **kwargs)
self.dimensions = 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 fd1698ea..aa55604a 100644
--- a/src/python/gudhi/tensorflow/lower_star_simplex_tree_layer.py
+++ b/src/python/gudhi/tensorflow/lower_star_simplex_tree_layer.py
@@ -49,7 +49,7 @@ class LowerStarSimplexTreeLayer(tf.keras.layers.Layer):
Parameters:
simplextree (gudhi.SimplexTree): underlying simplex tree. Its vertices MUST be named with integers from 0 to n = number of vertices
- dimensions (int): homology dimensions
+ dimensions (List[int]): homology dimensions
"""
super().__init__(dynamic=True, **kwargs)
self.dimensions = dimensions
diff --git a/src/python/gudhi/tensorflow/rips_layer.py b/src/python/gudhi/tensorflow/rips_layer.py
index da7087f6..472a418b 100644
--- a/src/python/gudhi/tensorflow/rips_layer.py
+++ b/src/python/gudhi/tensorflow/rips_layer.py
@@ -46,7 +46,7 @@ class RipsLayer(tf.keras.layers.Layer):
Parameters:
maximum_edge_length (float): maximum edge length for the Rips complex
- dimensions (int): homology dimensions
+ dimensions (List[int]): homology dimensions
"""
super().__init__(dynamic=True, **kwargs)
self.max_edge = maximum_edge_length