From 3e01470aa001cb2aedca3b66cbe2644e58d9c0f1 Mon Sep 17 00:00:00 2001
From: Andy Eschbacher <andy.eschbacher@gmail.com>
Date: Mon, 27 Jun 2016 10:07:22 -0400
Subject: [PATCH] add error catching

---
 .../crankshaft/segmentation/segmentation.py    | 18 ++++++++----------
 1 file changed, 8 insertions(+), 10 deletions(-)

diff --git a/src/py/crankshaft/crankshaft/segmentation/segmentation.py b/src/py/crankshaft/crankshaft/segmentation/segmentation.py
index d19eefd..5f11e04 100644
--- a/src/py/crankshaft/crankshaft/segmentation/segmentation.py
+++ b/src/py/crankshaft/crankshaft/segmentation/segmentation.py
@@ -40,17 +40,15 @@ def get_data(variable, feature_columns, query):
             prepared data, packaged into NumPy arrays
     """
 
-    columns  = ','.join(['array_agg("{col}") As "{col}"'.format(col=col) for col in feature_columns])
+    columns = ','.join(['array_agg("{col}") As "{col}"'.format(col=col) for col in feature_columns])
 
-    data = plpy.execute('''
-        SELECT array_agg("{variable}") As target,
-               {columns}
-        FROM ({query}) As a
-        '''.format(
-        variable = variable,
-        columns = columns,
-        query = query
-        ))
+    try:
+        data = plpy.execute('''SELECT array_agg("{variable}") As target, {columns} FROM ({query}) As a'''.format(
+            variable=variable,
+            columns=columns,
+            query=query))
+    except Exception, e:
+        plpy.error('failed to fetch data to construct model')
 
     target = np.array(data[0]['target'])