Fix leaf labels bug

Author: Krsto Proroković

unsupervised_bias_detection/cluster/_bahc.py

@@ -151,8 +151,8 @@ def fit(self, X, y):
         label_mapping = np.zeros(self.n_clusters_, dtype=np.uint32)
         label_mapping[leaf_labels] = np.arange(self.n_clusters_, dtype=np.uint32)
         self.labels_ = label_mapping[labels]
-        for i in range(self.n_clusters_):
-            leaves[i].label = leaf_labels[i]
+        for leaf in leaves:
+            leaf.label = label_mapping[leaf.label]
         return self
 
     def predict(self, X):