From 1511ecc67692773f016cc94190785faa33eab2d5 Mon Sep 17 00:00:00 2001 From: "D.J. Sutherland" Date: Wed, 13 May 2020 15:16:21 -0500 Subject: rename I to img in examples, since flake8 complains now (#176) --- examples/gromov/plot_gromov_barycenter.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'examples/gromov/plot_gromov_barycenter.py') diff --git a/examples/gromov/plot_gromov_barycenter.py b/examples/gromov/plot_gromov_barycenter.py index f6f031a..e2d88ba 100755 --- a/examples/gromov/plot_gromov_barycenter.py +++ b/examples/gromov/plot_gromov_barycenter.py @@ -84,9 +84,9 @@ def smacof_mds(C, dim, max_iter=3000, eps=1e-9): # The four distributions are constructed from 4 simple images -def im2mat(I): +def im2mat(img): """Converts and image to matrix (one pixel per line)""" - return I.reshape((I.shape[0] * I.shape[1], I.shape[2])) + return img.reshape((img.shape[0] * img.shape[1], img.shape[2])) square = pl.imread('../../data/square.png').astype(np.float64)[:, :, 2] -- cgit v1.2.3