Using numpy arrays instead of dataframes

Author: gAldeia

experiment/test_population.py

@@ -46,7 +46,7 @@ def test_population(ml):
     # Few samples to try to make it quick
     sample_idx = np.random.choice(np.arange(len(X_train)), size=10)
 
-    y_train = y_train[sample_idx]
+    y_train = y_train[sample_idx].values
     X_train = X_train.iloc[sample_idx]
 
     algorithm.est.fit(X_train, y_train)