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