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 | |
parent | b853e6a9db3dfc1f23d8f0b5101ca82dee686ecb (diff) |
[MRG] Gromov barycenter example bug (#425)
* bug solve
* releases.md
-rw-r--r-- | RELEASES.md | 2 | ||||
-rwxr-xr-x | examples/gromov/plot_gromov_barycenter.py | 3 |
2 files changed, 3 insertions, 2 deletions
diff --git a/RELEASES.md b/RELEASES.md index 9cfdd35..49475f2 100644 --- a/RELEASES.md +++ b/RELEASES.md @@ -32,6 +32,8 @@ roughly 2^31) (PR #381) - Fixed weak optimal transport docstring (Issue #404, PR #410) - Fixed error whith parameter `log=True`for `SinkhornLpl1Transport` (Issue #412, PR #413) +- Fixed a bug breaking an example where we would try to make an array of arrays of different shapes (Issue #424, PR #425) + ## 0.8.2 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
|