add error catching

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('''
+    try:
-        SELECT array_agg("{variable}") As target,
+        data = plpy.execute('''SELECT array_agg("{variable}") As target, {columns} FROM ({query}) As a'''.format(
-               {columns}
+            variable=variable,
-        FROM ({query}) As a
+            columns=columns,
-        '''.format(
+            query=query))
-        variable = variable,
+    except Exception, e:
-        columns = columns,
+        plpy.error('failed to fetch data to construct model')
-        query = query
-        ))
 
     target = np.array(data[0]['target'])