sklearn -> scikit-learn

setup.py

@@ -29,7 +29,7 @@ def get_version():
         'astropy>=2.0',
         'healpy >=1.10',
         'lmfit>=0.9.2',
-        'sklearn>=0.24.2']
+        'scikit-learn>=0.24.2']
 
 data_dir = 'AegeanTools/data'
 

tests/profile_cluster.py

@@ -0,0 +1,34 @@
+#! /usr/bin/env python
+"""
+Profile functions within the cluster module of AegeanTools
+"""
+
+from AegeanTools.cluster import regroup, regroup_vectorized
+from AegeanTools import catalogs
+
+import numpy as np
+import cProfile
+import pstats
+import io
+from pstats import SortKey
+
+
+def get_cat():
+    # create/load a catalogue for regrouping
+    table = catalogs.load_table('test_GLEAMX_comp.fits')[:1000]
+    srccat = catalogs.table_to_source_list(table)
+    return srccat
+
+
+def profile_regroup():
+    srccat = get_cat()
+    regroup(srccat, eps=np.sqrt(2))
+    return
+
+
+if __name__ == "__main__":
+    # introspect and run all the functions starting with 'test'
+    for f in dir():
+        if f.startswith('profile'):
+            print(f)
+            globals()[f]()