handle allocation of hdscan in CLUSTERINGS in case of exceptional import

openscilab · Dec 27, 2023 · 3d34029 · 3d34029
1 parent a79b86d
commit 3d34029
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_clusterings/test_clusterings.py b/tests/test_clusterings/test_clusterings.py
@@ -15,12 +15,13 @@
 from spectral_clustering import spectral_clustering
 from gaussian_mixture.gaussian_mixture import gaussian_mixture
 from hierarchical_clustering.agglomerative_clustering import agglomerative_clustering
+from pymilo.pymilo_param import SKLEARN_CLUSTERING_TABLE, NOT_SUPPORTED
 
 CLUSTERINGS = {
     "KMEANS": [kmeans],
     "AFFINITY_PROPAGATION": [affinity_propagation],
     "MEAN_SHIFT": [mean_shift],
-    "DBSCAN": [dbscan, hdbscan],
+    "DBSCAN": [dbscan, hdbscan if SKLEARN_CLUSTERING_TABLE["HDBSCAN"] != NOT_SUPPORTED else (None,"HDBSCAN")],
     "OPTICS": [optics],
     "SPECTRAL_CLUSTERING": [spectral_clustering],
     "GAUSSIAN_MIXTURE": [gaussian_mixture],