small update

rflamary · rflamary · commit 01f15d4ac4a9 · 2016-12-02T14:29:22.000+01:00
diff --git a/examples/plot_OTDA_classes.py b/examples/plot_OTDA_classes.py
@@ -11,6 +11,7 @@
 
 
 
+
 #%% parameters
 
 n=150 # nb samples in source and target datasets
diff --git 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
+    return x,y.astype(int)

Original file line number	Diff line number	Diff line change
`@@ -11,6 +11,7 @@`
`11`	`11`
`12`	`12`
`13`	`13`
	`14`	`+`
`14`	`15`	`#%% parameters`
`15`	`16`
`16`	`17`	`n=150 # nb samples in source and target datasets`