upd

Author: AlekseySh

docs/source/contents/datasets.rst

@@ -14,6 +14,7 @@ ImageBaseDataset
     :show-inheritance:
 
     .. automethod:: __init__
+    .. automethod:: __getitem__
     .. automethod:: visualize
 
 ImageLabeledDataset
@@ -25,6 +26,7 @@ ImageLabeledDataset
     .. automethod:: __init__
     .. automethod:: __getitem__
     .. automethod:: get_labels
+    .. automethod:: visualize
 
 ImageQueryGalleryLabeledDataset
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
@@ -36,6 +38,20 @@ ImageQueryGalleryLabeledDataset
     .. automethod:: __getitem__
     .. automethod:: get_query_ids
     .. automethod:: get_gallery_ids
+    .. automethod:: get_labels
+    .. automethod:: visualize
+
+ImageQueryGalleryDataset
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+.. autoclass:: oml.datasets.images.ImageQueryGalleryDataset
+    :undoc-members:
+    :show-inheritance:
+
+    .. automethod:: __init__
+    .. automethod:: __getitem__
+    .. automethod:: get_query_ids
+    .. automethod:: get_gallery_ids
+    .. automethod:: visualize
 
 EmbeddingPairsDataset
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

oml/datasets/images.py

@@ -133,7 +133,7 @@ def __init__(
         self.index_key = index_key
 
         if dataset_root is not None:
-            paths = list(map(lambda x: Path(dataset_root) / x), paths)  # type: ignore
+            paths = list(map(lambda x: Path(dataset_root) / x, paths))
 
         self._paths = list(map(str, paths))
         self._bboxes = bboxes