From 4709849bcc624d5e5c33755b997c56998f43d3e9 Mon Sep 17 00:00:00 2001
From: Kilian Fatras <kilianfatras@Kilians-MacBook-Air.local>
Date: Wed, 24 Jul 2019 18:46:44 +0200
Subject: corrected typos in barycenter examples

---
 examples/plot_barycenter_lp_vs_entropic.py | 7 ++++++-
 examples/plot_free_support_barycenter.py   | 2 +-
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/examples/plot_barycenter_lp_vs_entropic.py b/examples/plot_barycenter_lp_vs_entropic.py
index b82765e..d7c72d0 100644
--- a/examples/plot_barycenter_lp_vs_entropic.py
+++ b/examples/plot_barycenter_lp_vs_entropic.py
@@ -102,7 +102,7 @@ pl.tight_layout()
 problems.append([A, [bary_l2, bary_wass, bary_wass2]])
 
 ##############################################################################
-# Dirac Data
+# Stair Data
 # ----------
 
 #%% parameters
@@ -168,6 +168,11 @@ pl.legend()
 pl.title('Barycenters')
 pl.tight_layout()
 
+
+##############################################################################
+# Dirac Data
+# ----------
+
 #%% parameters
 
 a1 = np.zeros(n)
diff --git a/examples/plot_free_support_barycenter.py b/examples/plot_free_support_barycenter.py
index b6efc59..64b89e4 100644
--- a/examples/plot_free_support_barycenter.py
+++ b/examples/plot_free_support_barycenter.py
@@ -62,7 +62,7 @@ X = ot.lp.free_support_barycenter(measures_locations, measures_weights, X_init,
 pl.figure(1)
 for (x_i, b_i) in zip(measures_locations, measures_weights):
     color = np.random.randint(low=1, high=10 * N)
-    pl.scatter(x_i[:, 0], x_i[:, 1], s=b * 1000, label='input measure')
+    pl.scatter(x_i[:, 0], x_i[:, 1], s=b_i * 1000, label='input measure')
 pl.scatter(X[:, 0], X[:, 1], s=b * 1000, c='black', marker='^', label='2-Wasserstein barycenter')
 pl.title('Data measures and their barycenter')
 pl.legend(loc=0)
-- 
cgit v1.2.3