tweak test results from doc

1 files changed, 4 insertions, 5 deletions

diff --git a/src/python/gudhi/representations/vector_methods.py b/src/python/gudhi/representations/vector_methods.py
index 16e91812..d3b85636 100644
--- a/src/python/gudhi/representations/vector_methods.py
+++ b/src/python/gudhi/representations/vector_methods.py
@@ -614,18 +614,17 @@ class Atol(BaseEstimator, TransformerMixin):
     >>> b = np.array([[4, 2, 0], [4, 4, 0], [4, 0, 2]])
     >>> c = np.array([[3, 2, -1], [1, 2, -1]])
     >>> atol_vectoriser = Atol(quantiser=KMeans(n_clusters=2))
-    >>> atol_vectoriser.fit(X=[a, b, c])
-    >>> atol_vectoriser.centers
+    >>> atol_vectoriser.fit(X=[a, b, c]).centers
     array([[ 2.6       ,  2.8       , -0.4       ],
-       [ 2.        ,  0.66666667,  3.33333333]])
+           [ 2.        ,  0.66666667,  3.33333333]])
     >>> atol_vectoriser(a)
     array([0.58394704, 1.0769395 ])
     >>> atol_vectoriser(c)
     array([1.02816136, 0.23559623])
     >>> atol_vectoriser.transform(X=[a, b, c])
     array([[0.58394704, 1.0769395 ],
-       [1.04696684, 0.56203292],
-       [1.02816136, 0.23559623]])
+           [1.04696684, 0.56203292],
+           [1.02816136, 0.23559623]])
     """
     def __init__(self, quantiser, weighting_method="cloud", contrast="gaus"):
         """