Skip to content

Commit 62e12e4

Browse files
authored
[MRG] Gromov barycenter example bug (#425)
* bug solve * releases.md
1 parent b853e6a commit 62e12e4

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

RELEASES.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,8 @@ roughly 2^31) (PR #381)
3232
- Fixed weak optimal transport docstring (Issue #404, PR #410)
3333
- Fixed error whith parameter `log=True`for `SinkhornLpl1Transport` (Issue #412,
3434
PR #413)
35+
- Fixed a bug breaking an example where we would try to make an array of arrays of different shapes (Issue #424, PR #425)
36+
3537

3638
## 0.8.2
3739

examples/gromov/plot_gromov_barycenter.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -110,8 +110,7 @@ def im2mat(img):
110110
if shapes[nb][i, j] < 0.95:
111111
xs[nb].append([j, 8 - i])
112112

113-
xs = np.array([np.array(xs[0]), np.array(xs[1]),
114-
np.array(xs[2]), np.array(xs[3])])
113+
xs = [np.array(xs[s]) for s in range(S)]
115114

116115
##############################################################################
117116
# Barycenter computation

0 commit comments

Comments
 (0)