summaryrefslogtreecommitdiff
path: root/ot/datasets.py
diff options
context:
space:
mode:
authorRémi Flamary <remi.flamary@gmail.com>2016-12-01 16:21:27 +0100
committerRémi Flamary <remi.flamary@gmail.com>2016-12-01 16:21:27 +0100
commit0a7ce083728568d86554c7b3541ea72ff1d758e1 (patch)
tree4ed97a26bca3cb93ea97cfe6f11ef4f1fb59adc6 /ot/datasets.py
parente3c0d3e1251e70df25614f8dc458e46897d2efa7 (diff)
update barycenter demo
Diffstat (limited to 'ot/datasets.py')
-rw-r--r--ot/datasets.py2
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