From 0a7ce083728568d86554c7b3541ea72ff1d758e1 Mon Sep 17 00:00:00 2001
From: Rémi Flamary <remi.flamary@gmail.com>
Date: Thu, 1 Dec 2016 16:21:27 +0100
Subject: update barycenter demo

---
 ot/datasets.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'ot/datasets.py')

diff --git a/ot/datasets.py b/ot/datasets.py
index 8605691..5c1ef78 100644
--- a/ot/datasets.py
+++ b/ot/datasets.py
@@ -128,4 +128,4 @@ def get_data_classif(dataset,n,nz=.5,theta=0,**kwargs):
         y=0
         print("unknown dataset")
 
-    return x,y.astype(int)
\ No newline at end of file
+    return x,y
\ No newline at end of file
-- 
cgit v1.2.3