From a2b3733a1e52ef2b577ce01b18026e96de203500 Mon Sep 17 00:00:00 2001 From: Andy Eschbacher Date: Mon, 27 Jun 2016 09:58:42 -0400 Subject: [PATCH] fix improper line drop --- src/py/crankshaft/crankshaft/segmentation/segmentation.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/py/crankshaft/crankshaft/segmentation/segmentation.py b/src/py/crankshaft/crankshaft/segmentation/segmentation.py index da81842..d19eefd 100644 --- 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 # --------------------