correctly finds the number of columns

2018-01-09 14:28:20 -05:00
parent b16c73f7d0
commit b0e3f38f1e
1 changed files with 1 additions and 1 deletions
--- a/src/py/crankshaft/crankshaft/clustering/kmeans.py
+++ b/src/py/crankshaft/crankshaft/clustering/kmeans.py
@@ -97,7 +97,7 @@ def _extract_columns(data):
        data (list of dicts): result of the kmeans request
    """
    # number of columns minus rowid column
-    n_cols = len(data) - 1
+    n_cols = len(data[0]) - 1
    return np.array([data[0]['arr_col{0}'.format(i+1)]
                     for i in xrange(n_cols)],
                    dtype=float).T