diff --git a/mteb/evaluation/evaluators/model_classes.py b/mteb/evaluation/evaluators/model_classes.py index 7c984b634..60dea5638 100644 --- a/mteb/evaluation/evaluators/model_classes.py +++ b/mteb/evaluation/evaluators/model_classes.py @@ -343,12 +343,6 @@ def _full_corpus_search( query_embeddings, sub_corpus_embeddings ) - is_nan = torch.isnan(cos_scores) - if is_nan.sum() > 0: - raise ValueError( - f"NaN values detected in the similarity scores: {is_nan.sum()}" - ) - # get top-k values cos_scores_top_k_values, cos_scores_top_k_idx = torch.topk( cos_scores,