upd

Author: AlekseySh

oml/metrics/embeddings.py

@@ -212,7 +212,7 @@ def compute_metrics(self) -> TMetricsDict_ByLabels:  # type: ignore
         # note, here we do micro averaging
         metrics[self.overall_categories_key] = calc_retrieval_metrics(
             distances=self.distance_matrix,
-            gt_ids=self.mask_gt,
+            mask_gt=self.mask_gt,
             reduce=False,
             mask_to_ignore=None,  # we already applied it
             **args_retrieval_metrics,  # type: ignore
@@ -231,7 +231,7 @@ def compute_metrics(self) -> TMetricsDict_ByLabels:  # type: ignore
 
                 metrics[category] = calc_retrieval_metrics(
                     distances=self.distance_matrix[mask],  # type: ignore
-                    gt_ids=self.mask_gt[mask],  # type: ignore
+                    mask_gt=self.mask_gt[mask],  # type: ignore
                     reduce=False,
                     mask_to_ignore=None,  # we already applied it
                     **args_retrieval_metrics,  # type: ignore

tests/test_oml/test_functional/test_metrics/test_cmc_metric_old.py

@@ -13,7 +13,7 @@
 def cmc_score_count(distances: Tensor, mask_gt: Tensor, topk: int, mask_to_ignore: Optional[Tensor] = None) -> float:
     metrics = calc_retrieval_metrics(
         distances=distances,
-        gt_ids=mask_gt,
+        mask_gt=mask_gt,
         mask_to_ignore=mask_to_ignore,
         cmc_top_k=(topk,),
         map_top_k=tuple(),