fix improper line drop

2016-06-27 09:58:42 -04:00
parent a5e4ae99ce
commit a2b3733a1e
1 changed files with 1 additions and 2 deletions
--- a/src/py/crankshaft/crankshaft/segmentation/segmentation.py
+++ b/src/py/crankshaft/crankshaft/segmentation/segmentation.py
@@ -57,8 +57,7 @@ def get_data(variable, feature_columns, query):
    # put arrays into an n x m array of arrays
    features = np.column_stack([np.array(data[0][col], dtype=float) for col in feature_columns])

-    return replace_nan_with_mean(target),
-           replace_nan_with_mean(features)
+    return replace_nan_with_mean(target), replace_nan_with_mean(features)

 # High level interface
 # --------------------