minor fix to top_k_overlap

dilyabareeva · Jun 4, 2024 · b75edd9 · b75edd9
1 parent 0322da5
commit b75edd9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/metrics/unnamed/top_k_overlap.py b/src/metrics/unnamed/top_k_overlap.py
@@ -29,7 +29,7 @@ def compute(self, *args, **kwargs):
         return len(torch.unique(self.all_top_k_examples))
 
     def reset(self, *args, **kwargs):
-        self.all_top_k_examples = []
+        self.all_top_k_examples = torch.empty(0, top_k)
 
     def load_state_dict(self, state_dict: dict, *args, **kwargs):
         self.all_top_k_examples = state_dict["all_top_k_examples"]