summaryrefslogtreecommitdiff
path: root/examples
diff options
context:
space:
mode:
authorNathan Cassereau <84033440+ncassereau-idris@users.noreply.github.com>2022-12-20 14:06:50 +0100
committerGitHub <noreply@github.com>2022-12-20 14:06:50 +0100
commit62e12e4e2f260581d84f9614975a3e656f26eb82 (patch)
tree8bc001d287399accd941926a21554b391b83f0f0 /examples
parentb853e6a9db3dfc1f23d8f0b5101ca82dee686ecb (diff)
[MRG] Gromov barycenter example bug (#425)
* bug solve * releases.md
Diffstat (limited to 'examples')
-rwxr-xr-xexamples/gromov/plot_gromov_barycenter.py3
1 files changed, 1 insertions, 2 deletions
diff --git a/examples/gromov/plot_gromov_barycenter.py b/examples/gromov/plot_gromov_barycenter.py
index 7fe081f..08ec610 100755
--- a/examples/gromov/plot_gromov_barycenter.py
+++ b/examples/gromov/plot_gromov_barycenter.py
@@ -110,8 +110,7 @@ for nb in range(4):
if shapes[nb][i, j] < 0.95:
xs[nb].append([j, 8 - i])
-xs = np.array([np.array(xs[0]), np.array(xs[1]),
- np.array(xs[2]), np.array(xs[3])])
+xs = [np.array(xs[s]) for s in range(S)]
##############################################################################
# Barycenter computation