Rename deprecated keyword to

MolecularAI · Jan 23, 2024 · 2182894 · 2182894
1 parent 21455ad
commit 2182894
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/route_distances/clustering.py b/route_distances/clustering.py
@@ -41,7 +41,7 @@ def fixed_clustering(self, n_clusters: int, **kwargs: Any) -> np.ndarray:
         """
         if "linkage" not in kwargs:
             kwargs["linkage"] = "single"
-        kwargs["affinity"] = "precomputed"
+        kwargs["metric"] = "precomputed"
         kwargs["n_clusters"] = n_clusters
         model = AgglomerativeClustering(**kwargs)
         model.fit(self._distances)
@@ -59,7 +59,7 @@ def linkage_matrix(self, **kwargs: Any) -> np.ndarray:
         """
         if "linkage" not in kwargs:
             kwargs["linkage"] = "single"
-        kwargs["affinity"] = "precomputed"
+        kwargs["metric"] = "precomputed"
         kwargs["n_clusters"] = None
         kwargs["distance_threshold"] = 0.0
         model = AgglomerativeClustering(**kwargs)