From 0e8363ca13034537ca7bc64acd982f39b7a42123 Mon Sep 17 00:00:00 2001
From: Rémi Flamary <remi.flamary@gmail.com>
Date: Fri, 28 Oct 2016 09:14:19 +0200
Subject: doc sinkhorn

---
 examples/demo_OT_1D.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'examples')

diff --git a/examples/demo_OT_1D.py b/examples/demo_OT_1D.py
index fc2cbcf..6eaa2ff 100644
--- a/examples/demo_OT_1D.py
+++ b/examples/demo_OT_1D.py
@@ -8,7 +8,7 @@ Demo for 1D optimal transport
 import numpy as np
 import matplotlib.pylab as pl
 import ot
-
+from ot.datasets import get_1D_gauss as gauss
 
 
 #%% parameters
@@ -19,8 +19,8 @@ n=100 # nb bins
 x=np.arange(n,dtype=np.float64)
 
 # Gaussian distributions
-a=ot.datasets.get_1D_gauss(n,m=20,s=20) # m= mean, s= std
-b=ot.datasets.get_1D_gauss(n,m=60,s=60)
+a=gauss(n,m=20,s=20) # m= mean, s= std
+b=gauss(n,m=60,s=60)
 
 # loss matrix
 M=ot.dist(x.reshape((n,1)),x.reshape((n,1)))
-- 
cgit v1.2.3