From 01f15d4ac4a909ca7e71f6b8f6bca3edf88d7e47 Mon Sep 17 00:00:00 2001
From: Rémi Flamary <remi.flamary@gmail.com>
Date: Fri, 2 Dec 2016 14:29:22 +0100
Subject: small update

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

(limited to 'ot/datasets.py')

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