diff options
-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
|