diff options
author | Nathan Cassereau <84033440+ncassereau-idris@users.noreply.github.com> | 2022-12-20 14:06:50 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-12-20 14:06:50 +0100 |
commit | 62e12e4e2f260581d84f9614975a3e656f26eb82 (patch) | |
tree | 8bc001d287399accd941926a21554b391b83f0f0 /examples/gromov/plot_gromov_barycenter.py | |
parent | b853e6a9db3dfc1f23d8f0b5101ca82dee686ecb (diff) |
[MRG] Gromov barycenter example bug (#425)
* bug solve
* releases.md
Diffstat (limited to 'examples/gromov/plot_gromov_barycenter.py')
-rwxr-xr-x | examples/gromov/plot_gromov_barycenter.py | 3 |
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
|