From 29a0d810ed106357cdde503792fa4f1b589dd1bb Mon Sep 17 00:00:00 2001 From: mehak-sachdeva Date: Sun, 5 Feb 2017 20:17:28 -0500 Subject: [PATCH] data entry edits --- src/py/crankshaft/crankshaft/analysis_data_provider.py | 1 + src/py/crankshaft/test/test_segmentation.py | 1 - 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/src/py/crankshaft/crankshaft/analysis_data_provider.py b/src/py/crankshaft/crankshaft/analysis_data_provider.py index 8ef5929..9d373d9 100644 --- a/src/py/crankshaft/crankshaft/analysis_data_provider.py +++ b/src/py/crankshaft/crankshaft/analysis_data_provider.py @@ -77,6 +77,7 @@ class AnalysisDataProvider(object): "target": variable, "features": feature_columns} """ + columns = ', '.join(['array_agg("{col}") As "{col}"'.format(col=col) for col in params['features']]) query = ''' diff --git a/src/py/crankshaft/test/test_segmentation.py b/src/py/crankshaft/test/test_segmentation.py index a0f326d..faeafef 100644 --- a/src/py/crankshaft/test/test_segmentation.py +++ b/src/py/crankshaft/test/test_segmentation.py @@ -63,7 +63,6 @@ class SegmentationTest(unittest.TestCase): self.assertTrue(assert_array_equal(result, expectation)) def test_create_and_predict_segment(self): - from crankshaft.segmentation import Segmentation from numpy.testing import assert_array_equal n_samples = 1000