From 62e12e4e2f260581d84f9614975a3e656f26eb82 Mon Sep 17 00:00:00 2001
From: Nathan Cassereau <84033440+ncassereau-idris@users.noreply.github.com>
Date: Tue, 20 Dec 2022 14:06:50 +0100
Subject: [MRG] Gromov barycenter example bug (#425)

* bug solve

* releases.md
---
 examples/gromov/plot_gromov_barycenter.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

(limited to 'examples')

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
-- 
cgit v1.2.3