switch to torch.concat and torch.unique instead of list.append and set()

src/metrics/unnamed/top_k_overlap.py

@@ -15,18 +15,18 @@ def __init__(
     ):
         super().__init__(model, train_dataset, *args, **kwargs)
         self.top_k = top_k
-        self.all_top_k_examples = []
+        self.all_top_k_examples = torch.empty(0,top_k)
 
     def update(
         self,
         explanations: torch.Tensor,
         **kwargs,
     ):
         top_k_indices = torch.topk(explanations, self.top_k).indices
-        self.all_top_k_examples.append(top_k_indices)
+        self.all_top_k_examples=torch.concat((self.all_top_k_examples,top_k_indices), dim=0)
 
     def compute(self, *args, **kwargs):
-        return len(set(self.all_top_k_examples))
+        return len(torch.unique(self.all_top_k_examples))
 
     def reset(self, *args, **kwargs):
         self.all_top_k_examples = []