When there are no gts we raise warning instead of error

When there are no gts we raise warning instead of error

oml/retrieval/retrieval_results.py

@@ -1,3 +1,4 @@
+import warnings
 from copy import deepcopy
 from pprint import pformat
 from typing import Callable, List, Optional, Sequence, Tuple, Union
@@ -70,7 +71,7 @@ def __init__(
         if gt_ids is not None:
             assert len(distances) == len(gt_ids)
             if any(len(x) == 0 for x in gt_ids):
-                raise RuntimeError("Every query must have at least one relevant gallery id.")
+                warnings.warn("Some of the queries don't have available gts.")
 
         self._distances = tuple(distances)
         self._retrieved_ids = tuple(retrieved_ids)

tests/test_integrations/test_lightning/test_train_with_sequence.py

@@ -82,7 +82,7 @@ def test_invariants_in_validation_with_sequences_3() -> None:
     df_with_seq = df.copy()
     df_with_seq[SEQUENCE_COLUMN] = df_with_seq[LABELS_COLUMN]
 
-    with pytest.raises(RuntimeError):
+    with pytest.warns(UserWarning, match="Some of the queries don't have available gts."):
         validation(df)
         validation(df_with_seq)
 

tests/test_oml/test_retrieval_results/test_retrieval_results.py

@@ -159,7 +159,7 @@ def test_visualisation_for_different_number_of_retrieved_items() -> None:
 
 def test_retrieval_results_creation() -> None:
     # there is a query with no gt
-    with pytest.raises(RuntimeError):
+    with pytest.warns(UserWarning, match="Some of the queries don't have available gts."):
         RetrievalResults(
             distances=[torch.arange(3).float(), torch.arange(3).float()],
             retrieved_ids=[LongTensor([1, 0, 2]), LongTensor([4, 0, 1])],