[Fix] error in deleting not-matching gt values

src/similarity.py

@@ -35,7 +35,7 @@ def compute_pairwise_similarities(metric_name: str, probs: list[list[np.array]],
         metric = CAPA()
     elif metric_name == "CAPA (det.)":
         metric = CAPA(prob=False)
-        # Convert probabilities to one-hot
+        # Convert logits to one-hot
         probs = [[one_hot(p) for p in model_probs] for model_probs in probs]
     elif metric_name == "Error Consistency":
         probs = [[one_hot(p) for p in model_probs] for model_probs in probs]
@@ -48,25 +48,22 @@ def compute_pairwise_similarities(metric_name: str, probs: list[list[np.array]],
         for j in range(i, len(probs)):
             outputs_a = probs[i]
             outputs_b = probs[j]
-            gt_a = gts[i]
-            gt_b = gts[j]
+            gt_a = gts[i].copy()
+            gt_b = gts[j].copy()
 
             # Format softmax outputs
             if metric_name == "CAPA":
                 outputs_a = [softmax(logits) for logits in outputs_a]
                 outputs_b = [softmax(logits) for logits in outputs_b]
 
-            # Assert that the ground truth index is the same
-            indices_to_remove = []
-            if gt_a != gt_b:
-                for idx, (a, b) in enumerate(zip(gt_a, gt_b)):
-                    if a != b:
-                        indices_to_remove.append(idx)
+                # Remove indices where the ground truth differs
+                # (This code assumes gt_a and gt_b are lists of integers.)
+                indices_to_remove = [idx for idx, (a, b) in enumerate(zip(gt_a, gt_b)) if a != b]
                 for idx in sorted(indices_to_remove, reverse=True):
                     del outputs_a[idx]
                     del outputs_b[idx]
                     del gt_a[idx]
-                    del gt_b[idx] 
+                    del gt_b[idx]
 
             try:
                 similarities[i, j] = compute_similarity(metric, outputs_a, outputs_b, gt_a)