kaiko-eva 0.0.0.dev8__py3-none-any.whl → 0.0.2__py3-none-any.whl

{kaiko_eva-0.0.0.dev8.dist-info → kaiko_eva-0.0.2.dist-info}/RECORD

@@ -3,7 +3,8 @@ eva/__init__.py,sha256=bYBwklT7diG8NBIBDbpwjN4RUsvGv0ShWBXPxWgz404,518
 eva/__main__.py,sha256=kM5tQ0egTuBWixNLLx9QU-PpS2Bbs3zE3nYE6b2vWa0,282
 eva/__version__.py,sha256=YFR4oOlvPg0sS4Ni7GJ_vU42VTs5WiWp6odK7yH4TBY,611
 eva/core/__init__.py,sha256=AYlMZcH76B7I1lOa-E67u2o9DxsCwI4JMLCYXLk9oDQ,451
-eva/core/callbacks/__init__.py,sha256=tglEHfVB0NSaXenNLQAlyb2ufXiamxfR0o15W9a9t0k,110
+eva/core/callbacks/__init__.py,sha256=tHC6HJkLMKI5ThuSOuk3XLMvym4v2EtlDqrT91aAdmg,191
+eva/core/callbacks/config.py,sha256=-DRt20a2aF9Z9-7nZvbGBcOZ30qNf3ESf25EPRgRL1w,4267
 eva/core/callbacks/writers/__init__.py,sha256=GG8UXkbgNmpN1u_YIw-QysBGTSFm6C-P2XRtfnvVBrY,121
 eva/core/callbacks/writers/embeddings.py,sha256=q8Gd2_aDhhD7-QT5xAJbm8ikNjmHq_2DJosJ6WS5BOo,6752
 eva/core/callbacks/writers/typings.py,sha256=5AVIRAftqPTlLQ8s4ArEcMgLCyPnCZ9FFNk2yFppA1g,616
@@ -18,18 +19,31 @@ eva/core/data/datamodules/__init__.py,sha256=qZchYbgxo9lxYnGoqdk0C6MfS2IbF0WItO0
 eva/core/data/datamodules/call.py,sha256=jjj9w3UXYuQB-qyCcw1EZpRJW10OC1I3dvgvsuQWLck,940
 eva/core/data/datamodules/datamodule.py,sha256=dclC2YJAXUGEUpV9ZRWQS43-ksFIPgVeFudsyrj9kdc,3878
 eva/core/data/datamodules/schemas.py,sha256=EXnUPNd9Pj3RjnxJIzAcC2qp6TtBSvPDx28fV_ovWAA,1869
-eva/core/data/datasets/__init__.py,sha256=pMU-w6aQoRFgSwPB9GLXNilnsbwd6HqLlgw49e4rlj0,281
+eva/core/data/datasets/__init__.py,sha256=Y_18JE9sDtwEFdgPiKHf_ZqMj_1SUQtRZeiI2n7t1uA,387
 eva/core/data/datasets/base.py,sha256=NLZlxznB4SCYNf070OhfNJztaOpqwQWemwpGkFv_CA0,2005
-eva/core/data/datasets/classification/__init__.py,sha256=nhgeoK0chIvX9kzrp9yz40xha_g78o-R4AV7HbyfQOU,176
-eva/core/data/datasets/classification/embeddings.py,sha256=E2vYX5q0wDueIZwfB252aM4l9i8XNuf2BQ1lOJR32P4,5215
 eva/core/data/datasets/dataset.py,sha256=tA6Wd_7vqOE9GsukSWrgN9zaZKtKCHaE58SqIfWxWdg,124
+eva/core/data/datasets/embeddings/__init__.py,sha256=kxJtohGI6oP2lJiDVb61w7NMXuVD3np2B5wbuMIu6K4,357
+eva/core/data/datasets/embeddings/base.py,sha256=EXGa4qsXMA1LkdcZEdjnQq-ooSdw2VitjvpZLcTB8ug,4979
+eva/core/data/datasets/embeddings/classification/__init__.py,sha256=E2zyht5De7RnTVF-MRbteCzdUt4_Of2zJx3jolPpCdM,371
+eva/core/data/datasets/embeddings/classification/embeddings.py,sha256=d5s9f58oQKuRyzWGAg2RXFG41NjLWqu56O3_CI77S_8,2467
+eva/core/data/datasets/embeddings/classification/multi_embeddings.py,sha256=737KakBBui3zvMWKqoXWIG_qa5yaxhgtGP_iBCKnnPY,4343
 eva/core/data/samplers/__init__.py,sha256=WikBo1DemCx6o2vFfNwSwODlmCT2zWUXtCNwiWCVAFE,100
 eva/core/data/samplers/sampler.py,sha256=vrrXERWC67fjmTk_uwD7s9-8-rdhvnx7OlSipHE6sdY,119
-eva/core/data/transforms/__init__.py,sha256=ZdqOER1VMLu5nI1As0IuBPTS8ISw0EO4zeMZoio3G-c,157
+eva/core/data/transforms/__init__.py,sha256=n0TczmJSc9EjR6JezAZqlZIN4Gz_X3UBePbyDSC7JkE,308
 eva/core/data/transforms/dtype/__init__.py,sha256=r_LM_hdh_gTsrgh3shDTdMpu-lgQNHJ1yD6wY3omPyg,174
 eva/core/data/transforms/dtype/array.py,sha256=RDSkXlnSHSYyU_gv7vw33OZ7vhEy62PQGoE3htGGaqc,725
+eva/core/data/transforms/padding/__init__.py,sha256=AKSXa2dOhj45dTw81piPoCfDmIL0FPJUIxZ3HlG7KVM,138
+eva/core/data/transforms/padding/pad_2d_tensor.py,sha256=J4maGFmeQf9IHRxt5kU-6eI-Bvk12F_HVk8kR_omrnY,1185
+eva/core/data/transforms/sampling/__init__.py,sha256=BFKbvRjlZrwS0GcNrM54ZSWt6PrQARfFlXM1jJ-wpvo,149
+eva/core/data/transforms/sampling/sample_from_axis.py,sha256=Zbhp94lVa70WQKmSOKMTsOMe2c7wLqNZto7JqWhSdtI,1229
 eva/core/interface/__init__.py,sha256=chdpKXipxe1NP-Fgr_d9r6X1gMna0XiEa38waJ6FzTM,98
 eva/core/interface/interface.py,sha256=GzjneNHhTIEuLbydUG9cSmpHjJ4_IENGM-glN8RaRxY,2741
+eva/core/loggers/__init__.py,sha256=4YMLNlN9LnuKqhBI1R1keh69dmMD-2lcH3HKwwyn380,266
+eva/core/loggers/dummy.py,sha256=Y7ypH0ecSAIkkZ5LzTmNNEzlKkqeaHfUNMCDKVOg6D4,1204
+eva/core/loggers/experimental_loggers.py,sha256=JjW_SAABnNZyhJEy6kcPkPKDPyvuAj4sQTdn-tewm8U,204
+eva/core/loggers/log/__init__.py,sha256=cPqLKqIUhiTeDNG5ZUUOExDr76y4LBe4yXUvCxIQh0o,124
+eva/core/loggers/log/parameters.py,sha256=7qxa-ndtYb2_UkPZ4AlcHaGeUvEw2_9phCdco_OZ9Oo,1413
+eva/core/loggers/log/utils.py,sha256=k4Q7uKpAQctfDv0EEYPnPv6wt9LnckEeqGvbYSLfKO0,415
 eva/core/metrics/__init__.py,sha256=sTpNUbvgpKTd1IifzPVWmXVZ17PSJjcfEAlY_3fZP5U,558
 eva/core/metrics/average_loss.py,sha256=AyFOnCXBD5T62eSYf6eGAAJsqt8x-KaHgc8OLkCHjzE,1267
 eva/core/metrics/binary_balanced_accuracy.py,sha256=MabsXAtVfLqSaSIIpE0HIM6bo8uRszl6obueHI6vJi0,806
@@ -45,9 +59,9 @@ eva/core/metrics/structs/schemas.py,sha256=S6dTbz6YjxkNUIqWVd52KgpVx5JqNFqM4Xs7z
 eva/core/metrics/structs/typings.py,sha256=qJd-FiD2IhJgBeo8FyP0vpVUIH4RKb1k6zYvHtjUA04,388
 eva/core/models/__init__.py,sha256=yRLRKYuShhgQBWzV6sjzjThOqqNb9HRS48bMmAxEy-8,305
 eva/core/models/modules/__init__.py,sha256=QJWJ42BceXZBzDGgk5FHBcCaRrB9egTFKVF6gDsBYfM,255
-eva/core/models/modules/head.py,sha256=Rn9bJmSRWqNPnRl-YpvPZgtjbr6UNU83oHXQ1gEpAJE,4138
+eva/core/models/modules/head.py,sha256=sJ2RTNmOsqpCczajwgp6nHZ5qELhjEs8BhO36KyDG-g,3992
 eva/core/models/modules/inference.py,sha256=ih-0Rr2oNf2N6maiXPOW7XH5KVwUT1_MOxnJKOhJ1uQ,978
-eva/core/models/modules/module.py,sha256=0KquNWeblYg49S8AoV-Dpc65_ZIWl56ht5R7bKKXj7o,5918
+eva/core/models/modules/module.py,sha256=7mCzyvBNOWhvN8sNa91yB79iSBlJlYh9sypL37Nwdes,6836
 eva/core/models/modules/typings.py,sha256=fNoGsC_q1d9c2KauUC-f1psKrCFmfoeC8JJ_US_pOW0,521
 eva/core/models/modules/utils/__init__.py,sha256=pnbxlEhT87JimWNr-NSNCv7VNR-IyDi_A9qRWmvlzwQ,227
 eva/core/models/modules/utils/batch_postprocess.py,sha256=q1kC3pwSS7RyI76qunuvFP7RpUhEpUsR6xzjahJkQKQ,1915
@@ -64,48 +78,50 @@ eva/core/models/networks/wrappers/huggingface.py,sha256=81j0pcEx3DW6gR-81Fz6tZkJ
 eva/core/models/networks/wrappers/onnx.py,sha256=LZEGOpg1VYrB3wXMAA5IMfiKNTkOXQ50agHjTvYnnsU,1718
 eva/core/trainers/__init__.py,sha256=jhsKJF7HAae7EOiG3gKIAHH_h3dZlTE2JRcCHJmOzJc,208
 eva/core/trainers/_logging.py,sha256=gi4FqPy2GuVmh0WZY6mYwF7zMPvnoFA050B0XdCP6PU,2571
-eva/core/trainers/_recorder.py,sha256=_Vfp7Njh_9qP-SWbBGYp8solnfFgIUi2Z9pGLXt52WY,5652
+eva/core/trainers/_recorder.py,sha256=y6i5hfXftWjeV3eQHmMjUOkWumnZ2QNv_u275LLmvPA,7702
 eva/core/trainers/_utils.py,sha256=M3h8lVhUmkeSiEXpX9hRdMvThGFCnTP15gv-hd1CZkc,321
-eva/core/trainers/functional.py,sha256=pIeGXoO63Wh6n1mOYlBo5ACCteGuNV9pZhqxfN4RLSs,3775
-eva/core/trainers/trainer.py,sha256=j4rYWiG9COxBbZ6WIlpRPBE153XqlYK7eAWlRsZgljU,3261
+eva/core/trainers/functional.py,sha256=Ju0Eh98oxsj_7QBwxu3GddCD7FROEdQgpp9fdGS9sp4,4360
+eva/core/trainers/trainer.py,sha256=Vw_KhTyh-3YV5qo_XHxz9oy-v2PxrgoOWMeYi8-41R0,3949
 eva/core/utils/__init__.py,sha256=F1C69M9y7W8qh1J2k-X4frRHa7r1mPXewscC94fFYtk,58
 eva/core/utils/io/__init__.py,sha256=SAME0kuSvDE1DKFJwMBmnCkpDAy4ujXuRTSJsHNhwUI,112
 eva/core/utils/io/dataframe.py,sha256=CIHFowljH17waDkJ9YJVEVXAIcxMwoLjUgoBttiNk8w,509
 eva/core/utils/multiprocessing.py,sha256=PxUxMyvI62lghyWF46O5RNL-J7DUR2IrXSwdkbhC0ic,1383
 eva/core/utils/workers.py,sha256=hfx63M82qNg0Dwhre2tl53MnhtRsV7APaDONM9nhVB8,634
-eva/vision/__init__.py,sha256=Pp9FZomuqfbZdKSOmgSGfNYk8ISpXf7tLVBZfoFcybk,432
+eva/vision/__init__.py,sha256=Z9AuPmTO-i73pUtq3IkZzwRlY1E5xCE8IZiyl5S71TM,438
 eva/vision/data/__init__.py,sha256=aoKPmX8P2Q2k2W3nlq8vFU41FV6Sze-0SDuWtU-ETh4,111
-eva/vision/data/datasets/__init__.py,sha256=aV4qPqtlt0PnaGoxUW_xEwAr8b8ddkl_YE4_fAdavds,497
-eva/vision/data/datasets/_utils.py,sha256=5GAZEHn-VezxTXaW1jVZO5zvdVl1Vz8_5gV2qkoMu4s,1414
+eva/vision/data/datasets/__init__.py,sha256=0pGqI6m4zUEwHPFHcE_7z2MsDS7hG9xL8mnXvSPklOo,402
+eva/vision/data/datasets/_utils.py,sha256=epPcaYE4w2_LtUKLLQJh6qQxUNVBe22JA06k4WUerYQ,1430
 eva/vision/data/datasets/_validators.py,sha256=uPbbUNnftb8mYzsKVrF-ZX_xinB2zQkuQLFYMprVjhY,2099
-eva/vision/data/datasets/classification/__init__.py,sha256=I9vTkETzGnTNNvyRB96ut1YHx9ARmZVO0-0l3ZLWEAs,520
+eva/vision/data/datasets/classification/__init__.py,sha256=DZq9__B4D2x1fbQsG-SwcOR_KF8ZFgCtwTzhC16p63c,362
 eva/vision/data/datasets/classification/bach.py,sha256=_xuA4evV9jCI76bUKbzom4ECLKShCsd95S8PtvhRAH4,5637
 eva/vision/data/datasets/classification/base.py,sha256=zBqn8rQP59j1DEChf3rDXgyMtB_sbug8kPvgFCqZyl4,3060
 eva/vision/data/datasets/classification/crc.py,sha256=7RR0PJWnhLMa3AUB_F2XMYawF5gnCNbGMv25ejOEeNA,5875
 eva/vision/data/datasets/classification/mhist.py,sha256=yoDHZ2vqa26YKVvJ9t6aidOVGazGIwUD6F3o0zNsxjM,3257
 eva/vision/data/datasets/classification/patch_camelyon.py,sha256=CH9sveoMppNWPQHm4qPTONRSGqX3O8P3OYwMB6mO678,7253
-eva/vision/data/datasets/classification/total_segmentator.py,sha256=OkbqS41ykdUX0wGf6jSja5WzeeRmevUnH5alfcEQhwg,8069
 eva/vision/data/datasets/segmentation/__init__.py,sha256=byQCBHicM6mQkljHPllUqRvoFaJxHtPMKcyjPmK6dUM,249
-eva/vision/data/datasets/segmentation/base.py,sha256=JogXJ3KiOaUybAcyvoqjR4yjlBfVTt2Rt8OOAz32Jrc,3630
-eva/vision/data/datasets/segmentation/total_segmentator.py,sha256=NUh-NlrsTcUsbe3qLd_d481mok970bNF7zIdpAS7eks,8075
+eva/vision/data/datasets/segmentation/base.py,sha256=XSD-hl7UtqshzF9ulImlL7v11lT2rWhhro7K3dF0Ucs,2937
+eva/vision/data/datasets/segmentation/total_segmentator.py,sha256=xKGQN03ybOXQE1EtO9WRnAZ_8B40uXtaz2WliNFJ4Uc,8705
 eva/vision/data/datasets/structs.py,sha256=RaTDW-B36PumcR5gymhCiX-r8GiKqIFcjqoEEjjFyUE,389
 eva/vision/data/datasets/vision.py,sha256=hKKFMb65UJQzOyYm8FTGkOGBOinMRu7R8sOFMbCmQX4,1100
 eva/vision/data/transforms/__init__.py,sha256=cHnLwyx6biAjqstD4IDspVtM-_dv7GBrQG6x_0SM8MM,120
 eva/vision/data/transforms/common/__init__.py,sha256=ZHzpdr-THc9CgFFbAVMWUiZrUNUiHnCDM8GYhM7tMfU,138
 eva/vision/data/transforms/common/resize_and_crop.py,sha256=IkAeTOe5TxK_cHzFvS7yW8YUh27C-KjXqekL7pfcT9A,1485
+eva/vision/models/.DS_Store,sha256=CKEtWIsQcwe9BwZ8pyCnXleWpY_4ilCps_OuRG9ZGaM,8196
 eva/vision/models/__init__.py,sha256=v9JhyLdy38XUkA0JmNMzSxNYmKi7nWBrp_XYgM7dmTU,89
+eva/vision/models/networks/.DS_Store,sha256=1lFlJ5EFymdzGAUAaI30vcaaLHt3F1LwpG7xILf9jsM,6148
 eva/vision/models/networks/__init__.py,sha256=IDpFsocWtyfe28vR9yMmXPYzV9X2NSDrTx7ewH8u-XU,170
 eva/vision/models/networks/abmil.py,sha256=N1eH4fn1nXmgXurSQyQIxxonv7nsqeeuPWaQSHeltfs,6796
 eva/vision/models/networks/postprocesses/__init__.py,sha256=nWBuROKE77W9xfyAxmS6L9IgOaXjcB5Qpaw1ihHG64E,148
 eva/vision/models/networks/postprocesses/cls.py,sha256=AFRrCjSmKy0n14toKa-G-QVx3dh-H4zSx5myi_P2OFA,822
 eva/vision/utils/__init__.py,sha256=vaUovprE743SmyFH8l6uk4pYSWpI4zxn7lN0EwePTJI,96
+eva/vision/utils/convert.py,sha256=tP0ps_IK2BXkuUsmI0UXVrSbmDAbjfLtQt3AiFtT2j4,679
 eva/vision/utils/io/__init__.py,sha256=Aw2UxGO3nbUidroMlS-MMJUALjQVvfsOZ1ZhcENDwRo,310
 eva/vision/utils/io/_utils.py,sha256=JzOt7Frj6ScF_aNjFtfHBn4ROnl6NhUZucmQhLc4Cww,768
 eva/vision/utils/io/image.py,sha256=2jzeVFMvIRhuTkIrQeLyu0y8GttLp6rWRjO9I2uw-I8,1489
-eva/vision/utils/io/nifti.py,sha256=ph9w8dNNSsJG2wI3NJNPTLyWdz2S0i9jD068nHXVVJs,1510
+eva/vision/utils/io/nifti.py,sha256=Rfbz49IakkAMimN8VnHJxDgxqTVwxGfxWFrTMXLrbJc,1659
 eva/vision/utils/io/text.py,sha256=uECChKjeKi4KQ-NqdO7ywAFS_TOEp2DQ5QQcuG8cb-4,472
-kaiko_eva-0.0.0.dev8.dist-info/METADATA,sha256=M4MrBZhhW_RFJcR3J8kjSgRN7lauwINudUoEO1y6AO8,21567
-kaiko_eva-0.0.0.dev8.dist-info/WHEEL,sha256=N2J68yzZqJh3mI_Wg92rwhw0rtJDFpZj9bwQIMJgaVg,90
-kaiko_eva-0.0.0.dev8.dist-info/entry_points.txt,sha256=oqtS2Yt5EBY4saLyCBC3Zev3huCORKTKWyPovX7QR8g,73
-kaiko_eva-0.0.0.dev8.dist-info/licenses/LICENSE,sha256=e6AEzr7j_R-PYr2qLO-JwLn8y70jbVD3U2mxbRmwcI4,11338
-kaiko_eva-0.0.0.dev8.dist-info/RECORD,,
+kaiko_eva-0.0.2.dist-info/METADATA,sha256=R4SakGonLZsdyj2yEgkkOa1Z96INt2AfP6Gvak8eN68,23562
+kaiko_eva-0.0.2.dist-info/WHEEL,sha256=vnE8JVcI2Wz7GRKorsPArnBdnW2SWKWGow5gu5tHlRU,90
+kaiko_eva-0.0.2.dist-info/entry_points.txt,sha256=oqtS2Yt5EBY4saLyCBC3Zev3huCORKTKWyPovX7QR8g,73
+kaiko_eva-0.0.2.dist-info/licenses/LICENSE,sha256=e6AEzr7j_R-PYr2qLO-JwLn8y70jbVD3U2mxbRmwcI4,11338
+kaiko_eva-0.0.2.dist-info/RECORD,,

{kaiko_eva-0.0.0.dev8.dist-info → kaiko_eva-0.0.2.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: pdm-backend (2.1.8)
+Generator: pdm-backend (2.3.0)
 Root-Is-Purelib: true
 Tag: py3-none-any

eva/core/data/datasets/classification/__init__.py

@@ -1,5 +0,0 @@
-"""Classification datasets API."""
-
-from eva.core.data.datasets.classification.embeddings import EmbeddingsClassificationDataset
-
-__all__ = ["EmbeddingsClassificationDataset"]

eva/vision/data/datasets/classification/total_segmentator.py

@@ -1,213 +0,0 @@
-"""TotalSegmentator 2D segmentation dataset class."""
-
-import functools
-import os
-from glob import glob
-from typing import Callable, Dict, List, Literal, Tuple
-
-import numpy as np
-from torchvision.datasets import utils
-from typing_extensions import override
-
-from eva.vision.data.datasets import _utils, _validators, structs
-from eva.vision.data.datasets.classification import base
-from eva.vision.utils import io
-
-
-class TotalSegmentatorClassification(base.ImageClassification):
-    """TotalSegmentator multi-label classification dataset."""
-
-    _train_index_ranges: List[Tuple[int, int]] = [(0, 83)]
-    """Train range indices."""
-
-    _val_index_ranges: List[Tuple[int, int]] = [(83, 103)]
-    """Validation range indices."""
-
-    _n_slices_per_image: int = 20
-    """The amount of slices to sample per 3D CT scan image."""
-
-    _resources_full: List[structs.DownloadResource] = [
-        structs.DownloadResource(
-            filename="Totalsegmentator_dataset_v201.zip",
-            url="https://zenodo.org/records/10047292/files/Totalsegmentator_dataset_v201.zip",
-            md5="fe250e5718e0a3b5df4c4ea9d58a62fe",
-        ),
-    ]
-    """Resources for the full dataset version."""
-
-    _resources_small: List[structs.DownloadResource] = [
-        structs.DownloadResource(
-            filename="Totalsegmentator_dataset_small_v201.zip",
-            url="https://zenodo.org/records/10047263/files/Totalsegmentator_dataset_small_v201.zip",
-            md5="6b5524af4b15e6ba06ef2d700c0c73e0",
-        ),
-    ]
-    """Resources for the small dataset version."""
-
-    def __init__(
-        self,
-        root: str,
-        split: Literal["train", "val"] | None,
-        version: Literal["small", "full"] = "small",
-        download: bool = False,
-        image_transforms: Callable | None = None,
-        target_transforms: Callable | None = None,
-    ) -> None:
-        """Initialize dataset.
-
-        Args:
-            root: Path to the root directory of the dataset. The dataset will
-                be downloaded and extracted here, if it does not already exist.
-            split: Dataset split to use. If None, the entire dataset is used.
-            version: The version of the dataset to initialize.
-            download: Whether to download the data for the specified split.
-                Note that the download will be executed only by additionally
-                calling the :meth:`prepare_data` method and if the data does not
-                exist yet on disk.
-            image_transforms: A function/transform that takes in an image
-                and returns a transformed version.
-            target_transforms: A function/transform that takes in the target
-                and transforms it.
-        """
-        super().__init__(
-            image_transforms=image_transforms,
-            target_transforms=target_transforms,
-        )
-
-        self._root = root
-        self._split = split
-        self._version = version
-        self._download = download
-
-        self._samples_dirs: List[str] = []
-        self._indices: List[int] = []
-
-    @functools.cached_property
-    @override
-    def classes(self) -> List[str]:
-        def get_filename(path: str) -> str:
-            """Returns the filename from the full path."""
-            return os.path.basename(path).split(".")[0]
-
-        first_sample_labels = os.path.join(
-            self._root, self._samples_dirs[0], "segmentations", "*.nii.gz"
-        )
-        return sorted(map(get_filename, glob(first_sample_labels)))
-
-    @property
-    @override
-    def class_to_idx(self) -> Dict[str, int]:
-        return {label: index for index, label in enumerate(self.classes)}
-
-    @override
-    def filename(self, index: int) -> str:
-        sample_dir = self._samples_dirs[self._indices[index]]
-        return os.path.join(sample_dir, "ct.nii.gz")
-
-    @override
-    def prepare_data(self) -> None:
-        if self._download:
-            self._download_dataset()
-        _validators.check_dataset_exists(self._root, True)
-
-    @override
-    def configure(self) -> None:
-        self._samples_dirs = self._fetch_samples_dirs()
-        self._indices = self._create_indices()
-
-    @override
-    def validate(self) -> None:
-        _validators.check_dataset_integrity(
-            self,
-            length=1660 if self._split == "train" else 400,
-            n_classes=117,
-            first_and_last_labels=("adrenal_gland_left", "vertebrae_T9"),
-        )
-
-    @override
-    def __len__(self) -> int:
-        return len(self._indices) * self._n_slices_per_image
-
-    @override
-    def load_image(self, index: int) -> np.ndarray:
-        image_path = self._get_image_path(index)
-        slice_index = self._get_sample_slice_index(index)
-        image_array = io.read_nifti_slice(image_path, slice_index)
-        return image_array.repeat(3, axis=2)
-
-    @override
-    def load_target(self, index: int) -> np.ndarray:
-        masks = self._load_masks(index)
-        targets = [1 in masks[..., mask_index] for mask_index in range(masks.shape[-1])]
-        return np.asarray(targets, dtype=np.int64)
-
-    def _load_masks(self, index: int) -> np.ndarray:
-        """Returns the `index`'th target mask sample."""
-        masks_dir = self._get_masks_dir(index)
-        slice_index = self._get_sample_slice_index(index)
-        mask_paths = (os.path.join(masks_dir, label + ".nii.gz") for label in self.classes)
-        masks = [io.read_nifti_slice(path, slice_index) for path in mask_paths]
-        return np.concatenate(masks, axis=-1)
-
-    def _get_masks_dir(self, index: int) -> str:
-        """Returns the directory of the corresponding masks."""
-        sample_dir = self._get_sample_dir(index)
-        return os.path.join(self._root, sample_dir, "segmentations")
-
-    def _get_image_path(self, index: int) -> str:
-        """Returns the corresponding image path."""
-        sample_dir = self._get_sample_dir(index)
-        return os.path.join(self._root, sample_dir, "ct.nii.gz")
-
-    def _get_sample_dir(self, index: int) -> str:
-        """Returns the corresponding sample directory."""
-        sample_index = self._indices[index // self._n_slices_per_image]
-        return self._samples_dirs[sample_index]
-
-    def _get_sample_slice_index(self, index: int) -> int:
-        """Returns the corresponding slice index."""
-        image_path = self._get_image_path(index)
-        total_slices = io.fetch_total_nifti_slices(image_path)
-        slice_indices = np.linspace(0, total_slices - 1, num=self._n_slices_per_image, dtype=int)
-        return slice_indices[index % self._n_slices_per_image]
-
-    def _fetch_samples_dirs(self) -> List[str]:
-        """Returns the name of all the samples of all the splits of the dataset."""
-        sample_filenames = [
-            filename
-            for filename in os.listdir(self._root)
-            if os.path.isdir(os.path.join(self._root, filename))
-        ]
-        return sorted(sample_filenames)
-
-    def _create_indices(self) -> List[int]:
-        """Builds the dataset indices for the specified split."""
-        split_index_ranges = {
-            "train": self._train_index_ranges,
-            "val": self._val_index_ranges,
-            None: [(0, 103)],
-        }
-        index_ranges = split_index_ranges.get(self._split)
-        if index_ranges is None:
-            raise ValueError("Invalid data split. Use 'train', 'val' or `None`.")
-
-        return _utils.ranges_to_indices(index_ranges)
-
-    def _download_dataset(self) -> None:
-        """Downloads the dataset."""
-        dataset_resources = {
-            "small": self._resources_small,
-            "full": self._resources_full,
-            None: (0, 103),
-        }
-        resources = dataset_resources.get(self._version)
-        if resources is None:
-            raise ValueError("Invalid data version. Use 'small' or 'full'.")
-
-        for resource in resources:
-            utils.download_and_extract_archive(
-                resource.url,
-                download_root=self._root,
-                filename=resource.filename,
-                remove_finished=True,
-            )