replace plpy method colnames

2016-11-15 11:19:15 +01:00 · 2016-11-15 11:19:15 +01:00 · 0867e69d1f
commit 0867e69d1f
parent cbe8571546
1 changed files with 3 additions and 1 deletions
--- a/src/py/crankshaft/crankshaft/clustering/kmeans.py
+++ b/src/py/crankshaft/crankshaft/clustering/kmeans.py
@ -92,7 +92,9 @@ def _extract_columns(db_resp, id_col_name):
        id_col_name (string): name of column which has the row id (not a
                              feature of the analysis)
    """
-    return np.array([db_resp[0][c] for c in db_resp.colnames()
+    keys = [k for k in db_resp[0].keys()]
+
+    return np.array([db_resp[0][c] for c in keys
                     if c != id_col_name],
                    dtype=float).T