1 files changed, 3 insertions, 3 deletions
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