make numpy with global scope in module

2016-10-13 15:00:28 +00:00
parent a95423174c
commit 5d2a1881b1
1 changed files with 1 additions and 2 deletions
--- a/src/py/crankshaft/crankshaft/clustering/kmeans.py
+++ b/src/py/crankshaft/crankshaft/clustering/kmeans.py
@@ -1,6 +1,6 @@
 from sklearn.cluster import KMeans
 import plpy
-
+import numpy as np

 def kmeans(query, no_clusters, no_init=20):
    """
@@ -39,7 +39,6 @@ def kmeans_nonspatial(query, colnames, num_clusters=5,
        num_clusters (int): number of clusters (greater than zero)
        id_col (string): name of the input id_column
    """
-    import numpy as np
    out_id_colname = 'rowids'
    # TODO: need a random seed?