tonic 1.5.1.dev19__tar.gz → 1.5.1.dev20__tar.gz

{tonic-1.5.1.dev19 → tonic-1.5.1.dev20}/.binder/requirements.txt

@@ -10,4 +10,3 @@ expelliarmus
 torchvision
 ipywidgets
 matplotlib
-torchdata

{tonic-1.5.1.dev19 → tonic-1.5.1.dev20}/ChangeLog

@@ -1,7 +1,8 @@
 CHANGES
 =======
 
-* add NeRDD dataset
+* exclude prototype from docs
+* remove prototype code because torchdata removed support for it
 * Revert bugfix
 * Added documentation for NTIDIGITS18
 * Added support for NTIDIGITS18 Dataset

{tonic-1.5.1.dev19 → tonic-1.5.1.dev20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tonic
-Version: 1.5.1.dev19
+Version: 1.5.1.dev20
 Summary: Neuromorphic datasets and transformations.
 Home-page: UNKNOWN
 Author: The Neuromorphs of Telluride

{tonic-1.5.1.dev19 → tonic-1.5.1.dev20}/docs/about/about.rst

@@ -3,6 +3,5 @@ About
 
 .. toctree::
    info
-   prototype
    release_notes
    

{tonic-1.5.1.dev19 → tonic-1.5.1.dev20}/docs/datasets.rst

@@ -45,28 +45,19 @@ Pose estimation, visual odometry, SLAM
     MVSEC
     TUMVIE
     VPR
-
-Object tracking
+    
+Star tracking
 -------------------
 .. autosummary::
     :toctree: generated/
     :template: class_dataset.rst
 
-    ThreeET_Eyetracking
-    NeRDD
     EBSSA
 
-
-.. currentmodule:: tonic.prototype.datasets
-
-Prototype iterable datasets
----------------------------
+Eye tracking
+-------------------
 .. autosummary::
     :toctree: generated/
     :template: class_dataset.rst
 
-    NMNIST
-    NCARS
-    STMNIST
-    Gen1AutomotiveDetection
-    Gen4AutomotiveDetectionMini
+    ThreeET_Eyetracking

{tonic-1.5.1.dev19 → tonic-1.5.1.dev20}/test/torch_requirements.txt

@@ -2,4 +2,3 @@
 torch==2.3.0
 torchaudio==2.3.0
 torchvision==0.18.0
-torchdata<=0.8.0

{tonic-1.5.1.dev19 → tonic-1.5.1.dev20}/tonic/__init__.py

@@ -10,10 +10,5 @@ from .cached_dataset import (
 from .dataset import Dataset
 from .sliced_dataset import SlicedDataset
 
-try:
-    from . import prototype
-except (ImportError, NameError):
-    pass
-
 all = "__version__"
 __version__ = VersionInfo("tonic").release_string()

{tonic-1.5.1.dev19 → tonic-1.5.1.dev20}/tonic/datasets/__init__.py

@@ -9,7 +9,6 @@ from .ntidigits18 import NTIDIGITS18
 from .hsd import SHD, SSC
 from .mvsec import MVSEC
 from .ncaltech101 import NCALTECH101
-from .nerdd import NERDD
 from .nmnist import NMNIST
 from .pokerdvs import POKERDVS
 from .s_mnist import SMNIST
@@ -26,7 +25,6 @@ __all__ = [
     "EBSSA",
     "MVSEC",
     "NCALTECH101",
-    "NERDD",
     "NMNIST",
     "POKERDVS",
     "SHD",

{tonic-1.5.1.dev19 → tonic-1.5.1.dev20}/tonic.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tonic
-Version: 1.5.1.dev19
+Version: 1.5.1.dev20
 Summary: Neuromorphic datasets and transformations.
 Home-page: UNKNOWN
 Author: The Neuromorphs of Telluride

{tonic-1.5.1.dev19 → tonic-1.5.1.dev20}/tonic.egg-info/SOURCES.txt

@@ -33,7 +33,6 @@ docs/_templates/class_dataset.rst
 docs/_templates/class_transform.rst
 docs/about/about.rst
 docs/about/info.rst
-docs/about/prototype.rst
 docs/about/release_notes.rst
 docs/gallery/README.rst
 docs/gallery/representations/README.rst
@@ -86,7 +85,6 @@ docs/tutorials/large_datasets.ipynb
 docs/tutorials/slicing.ipynb
 docs/tutorials/tutorials.rst
 test/dataset_utils.py
-test/prototype_dataset_utils.py
 test/requirements.txt
 test/test_audio_augmentations.py
 test/test_audio_transforms.py
@@ -96,7 +94,6 @@ test/test_chained_transforms.py
 test/test_datasets.py
 test/test_dsec.py
 test/test_io.py
-test/test_prototype_datasets.py
 test/test_representations.py
 test/test_sliced_dataset.py
 test/test_slicers.py
@@ -139,7 +136,6 @@ tonic/datasets/ebssa.py
 tonic/datasets/hsd.py
 tonic/datasets/mvsec.py
 tonic/datasets/ncaltech101.py
-tonic/datasets/nerdd.py
 tonic/datasets/nmnist.py
 tonic/datasets/ntidigits18.py
 tonic/datasets/pokerdvs.py
@@ -163,15 +159,4 @@ tonic/functional/to_bina_rep.py
 tonic/functional/to_frame.py
 tonic/functional/to_timesurface.py
 tonic/functional/to_voxel_grid.py
-tonic/functional/uniform_noise.py
-tonic/prototype/README.md
-tonic/prototype/__init__.py
-tonic/prototype/slicers.py
-tonic/prototype/datasets/__init__.py
-tonic/prototype/datasets/ncars.py
-tonic/prototype/datasets/nmnist.py
-tonic/prototype/datasets/prophesee.py
-tonic/prototype/datasets/stmnist.py
-tonic/prototype/datasets/utils/__init__.py
-tonic/prototype/datasets/utils/_dataset.py
-tonic/prototype/datasets/utils/_utils.py
+tonic/functional/uniform_noise.py

tonic-1.5.1.dev20/tonic.egg-info/pbr.json

@@ -0,0 +1 @@
+{"git_version": "3a8e1e2", "is_release": false}

tonic-1.5.1.dev19/docs/about/prototype.rst

@@ -1,6 +0,0 @@
-Prototype Tonic v2
-==================
-
-Lately torchvision is making efforts to change its datasets to a new format of iterable datasets. That means that the size of a dataset is not necessarily fixed upfront and samples can be streamed individually. As the sizes of datasets grow, it becomes infeasible to keep all samples in memory, sometimes even the list of file paths to all samples. This also quickly is becoming a problem for event-based data, as with increasing spatial sizes of event cameras the amount of events per second increases drastically.
-
-To address this issue, PyTorch created a new package `torchdata <https://pytorch.org/data/main/index.html>`_ which focuses on separating transformation logic (such as augementations) from loading functionality (batching / multithreading). Tonic will also take advantage of these newest developments and will preferrably implement new datasets in the new iterable style. 

tonic-1.5.1.dev19/test/prototype_dataset_utils.py

@@ -1,86 +0,0 @@
-import os
-import shutil
-import unittest
-from typing import Any, Dict, Union
-from unittest.mock import patch
-
-import numpy as np
-
-# On Unix it yields "/tmp", on Findus "C:\\tmp".
-# Actually, on Windows the temporary folder is in C:\\Users\<username>\AppData\Local\Temp (or C:\\System32\Temp, I do not remember). I do not know how to solve this.
-TMP_DIR = os.path.join(os.path.abspath(os.sep), "tmp")
-
-
-class DatasetTestCase(unittest.TestCase):
-    DATASET_CLASS = None
-    FEATURE_TYPES = None
-
-    _CHECK_FUNCTIONS = {"check_exists"}
-    _DOWNLOAD_EXTRACT_FUNCTIONS = {}
-
-    def inject_fake_data(
-        self, tmpdir: str, config: Dict[str, Any]
-    ) -> Union[int, Dict[str, Any]]:
-        """Inject fake data for dataset into a temporary directory.
-
-        During the creation of the dataset the download and extract logic is disabled. Thus, the fake data injected
-        here needs to resemble the raw data, i.e. the state of the dataset directly after the files are downloaded and
-        potentially extracted.
-        Args:
-            tmpdir (str): Path to a temporary directory. For most cases this acts as root directory for the dataset
-                to be created and in turn also for the fake data injected here.
-            config (Dict[str, Any]): Configuration that will be passed to the dataset constructor. It provides at least
-                fields for all dataset parameters with default values.
-        Needs to return one of the following:
-            1. (int): Number of examples in the dataset to be created, or
-            2. (Dict[str, Any]): Additional information about the injected fake data. Must contain the field
-                ``"num_examples"`` that corresponds to the number of examples in the dataset to be created.
-        """
-        raise NotImplementedError(
-            "You need to provide fake data in order for the tests to run."
-        )
-
-    def create_dataset(self, inject_fake_data: bool = True, **kwargs: Any):
-        self.inject_fake_data(TMP_DIR)
-
-        if inject_fake_data:
-            with patch.object(self.DATASET_CLASS, "_check_exists", return_value=True):
-                dataset = self.DATASET_CLASS(**self.KWARGS)
-        else:
-            dataset = self.DATASET_CLASS(**self.KWARGS)
-        return dataset
-
-    def _patch_checks(self):
-        return {
-            patch.object(self.DATASET_CLASS, function, return_value=True)
-            for function in self._CHECK_FUNCTIONS
-        }
-
-    #     def test_download_started(self):
-    #         with pytest.raises((FileNotFoundError, RuntimeError)):
-    #             dataset, info = self.create_dataset(inject_fake_data=False)
-
-    def test_feature_types(self):
-        dataset = self.create_dataset()
-        data, target = next(iter(dataset))
-
-        if type(data) != tuple:
-            data = (data,)
-        if type(target) != tuple:
-            target = (target,)
-        assert len(data) == len(self.FEATURE_TYPES)
-        assert len(target) == len(self.TARGET_TYPES)
-
-        for (data_piece, feature_type) in zip(data, self.FEATURE_TYPES):
-            if type(data_piece) == np.ndarray:
-                assert data_piece.dtype == feature_type
-            else:
-                assert type(data_piece) == feature_type
-
-    @classmethod
-    def setUpClass(cls):
-        cls.KWARGS.update({"root": TMP_DIR})
-        shutil.rmtree(
-            os.path.join(TMP_DIR, cls.DATASET_CLASS.__name__), ignore_errors=True
-        )
-        super().setUpClass()

tonic-1.5.1.dev19/test/test_prototype_datasets.py

@@ -1,300 +0,0 @@
-import shutil
-from pathlib import Path
-
-import numpy as np
-import prototype_dataset_utils as dataset_utils
-
-import tonic.prototype.datasets as datasets
-
-PATH_TO_TEST_DATA = Path(".", "test", "test_data")
-
-########
-# NMNIST
-########
-
-
-def create_nmnist_zip_archive(tmpdir: str, folder: str, filename: str):
-    """This function recreates the same structure that we expect in the dataset:
-
-    [Train,Test]----!                 !---0/sample.bin                 !---1/sample.bin
-    !---                 !---9/sample.bin
-    """
-    filename = filename.split(".")[0]  # Stripping the .zip extension.
-    folder_path = Path(tmpdir, "NMNIST", folder)
-    # We create a folder for each label.
-    for i in range(10):
-        destination_path = Path(folder_path, str(i))
-        destination_path.mkdir(parents=True, exist_ok=True)
-        shutil.copyfile(
-            PATH_TO_TEST_DATA / "sample_nmnist.bin",
-            destination_path / "sample_nmnist.bin",
-        )
-    # We compress the archive.
-    shutil.make_archive(
-        base_name=Path(tmpdir, "NMNIST", filename),
-        format="zip",
-        root_dir=folder_path.parent,
-        base_dir=folder,
-    )
-    # We remove the folder previously created.
-    shutil.rmtree(folder_path)
-
-
-class NMNISTTestCase_Train_Uncompressed_AllSaccades(dataset_utils.DatasetTestCase):
-    DATASET_CLASS = datasets.NMNIST
-    FEATURE_TYPES = (datasets.NMNIST._DTYPE,)
-    TARGET_TYPES = (int,)
-    KWARGS = {"train": True, "keep_compressed": False, "first_saccade_only": False}
-
-    def inject_fake_data(self, tmpdir):
-        create_nmnist_zip_archive(
-            tmpdir, datasets.NMNIST._TRAIN_FOLDER, datasets.NMNIST._TRAIN_FILENAME
-        )
-
-
-class NMNISTTestCase_Train_Compressed_AllSaccades(dataset_utils.DatasetTestCase):
-    DATASET_CLASS = datasets.NMNIST
-    FEATURE_TYPES = (datasets.NMNIST._DTYPE,)
-    TARGET_TYPES = (int,)
-    KWARGS = {"train": True, "keep_compressed": True, "first_saccade_only": False}
-
-    def inject_fake_data(self, tmpdir):
-        create_nmnist_zip_archive(
-            tmpdir, datasets.NMNIST._TRAIN_FOLDER, datasets.NMNIST._TRAIN_FILENAME
-        )
-
-
-class NMNISTTestCase_Train_Uncompressed_FirstSaccade(dataset_utils.DatasetTestCase):
-    DATASET_CLASS = datasets.NMNIST
-    FEATURE_TYPES = (datasets.NMNIST._DTYPE,)
-    TARGET_TYPES = (int,)
-    KWARGS = {"train": True, "keep_compressed": False, "first_saccade_only": True}
-
-    def inject_fake_data(self, tmpdir):
-        create_nmnist_zip_archive(
-            tmpdir, datasets.NMNIST._TRAIN_FOLDER, datasets.NMNIST._TRAIN_FILENAME
-        )
-
-
-class NMNISTTestCase_Test_Uncompressed_AllSaccades(dataset_utils.DatasetTestCase):
-    DATASET_CLASS = datasets.NMNIST
-    FEATURE_TYPES = (datasets.NMNIST._DTYPE,)
-    TARGET_TYPES = (int,)
-    KWARGS = {"train": False, "keep_compressed": False, "first_saccade_only": False}
-
-    def inject_fake_data(self, tmpdir):
-        create_nmnist_zip_archive(
-            tmpdir, datasets.NMNIST._TEST_FOLDER, datasets.NMNIST._TEST_FILENAME
-        )
-
-
-class NMNISTTestCase_Test_Compressed_AllSaccades(dataset_utils.DatasetTestCase):
-    DATASET_CLASS = datasets.NMNIST
-    FEATURE_TYPES = (datasets.NMNIST._DTYPE,)
-    TARGET_TYPES = (int,)
-    KWARGS = {"train": False, "keep_compressed": True, "first_saccade_only": False}
-
-    def inject_fake_data(self, tmpdir):
-        create_nmnist_zip_archive(
-            tmpdir, datasets.NMNIST._TEST_FOLDER, datasets.NMNIST._TEST_FILENAME
-        )
-
-
-class NMNISTTestCase_Test_Uncompressed_FirstSaccade(dataset_utils.DatasetTestCase):
-    DATASET_CLASS = datasets.NMNIST
-    FEATURE_TYPES = (datasets.NMNIST._DTYPE,)
-    TARGET_TYPES = (int,)
-    KWARGS = {"train": False, "keep_compressed": False, "first_saccade_only": True}
-
-    def inject_fake_data(self, tmpdir):
-        create_nmnist_zip_archive(
-            tmpdir, datasets.NMNIST._TEST_FOLDER, datasets.NMNIST._TEST_FILENAME
-        )
-
-
-##########
-# STMNIST
-##########
-
-
-def create_stmnist_zip_archive(tmpdir: str, filename: str):
-    """This function recreates the same structure that we expect in the dataset:
-    data_submission/----!
-
-    !---0/sample.mat !---1/sample.mat !--- !---9/sample.mat
-    """
-    folder_path = Path(tmpdir, "STMNIST", "data_submission")
-    # We create a folder for each label.
-    for i in range(10):
-        destination_path = Path(folder_path, str(i))
-        destination_path.mkdir(parents=True, exist_ok=True)
-        shutil.copyfile(
-            PATH_TO_TEST_DATA / "sample_stmnist.mat",
-            destination_path / "sample_stmnist.mat",
-        )
-    # We compress the archive.
-    shutil.make_archive(
-        base_name=Path(tmpdir, "STMNIST", filename),
-        format="zip",
-        base_dir=folder_path,
-        root_dir=folder_path,
-    )
-
-
-class STMNISTTestCase_Uncompressed(dataset_utils.DatasetTestCase):
-    DATASET_CLASS = datasets.STMNIST
-    FEATURE_TYPES = (datasets.STMNIST._DTYPE,)
-    TARGET_TYPES = (int,)
-    KWARGS = {"keep_compressed": False}
-
-    def inject_fake_data(self, tmpdir):
-        create_stmnist_zip_archive(tmpdir, datasets.STMNIST.__name__)
-
-
-class STMNISTTestCase_Compressed(dataset_utils.DatasetTestCase):
-    DATASET_CLASS = datasets.STMNIST
-    FEATURE_TYPES = (datasets.STMNIST._DTYPE,)
-    TARGET_TYPES = (int,)
-    KWARGS = {"keep_compressed": True}
-
-    def inject_fake_data(self, tmpdir):
-        create_stmnist_zip_archive(tmpdir, datasets.STMNIST.__name__)
-
-
-#######
-# NCARS
-#######
-
-
-def create_ncars_folder(tmpdir, folder_name):
-    for label_folder_name in ("background", "cars"):
-        test_dir = Path(tmpdir, "NCARS", folder_name, label_folder_name)
-        test_dir.mkdir(parents=True, exist_ok=True)
-        shutil.copyfile(
-            PATH_TO_TEST_DATA / "sample_ncars.dat",
-            test_dir / "sample_ncars.dat",
-        )
-
-
-class NCARSTestCase_Train(dataset_utils.DatasetTestCase):
-    DATASET_CLASS = datasets.NCARS
-    FEATURE_TYPES = (datasets.NCARS._DTYPE,)
-    TARGET_TYPES = (int,)
-    KWARGS = {"train": True}
-
-    def inject_fake_data(self, tmpdir):
-        create_ncars_folder(tmpdir, datasets.NCARS._TRAIN_PATH)
-
-
-class NCARSTestCase_Test(dataset_utils.DatasetTestCase):
-    DATASET_CLASS = datasets.NCARS
-    FEATURE_TYPES = (datasets.NCARS._DTYPE,)
-    TARGET_TYPES = (int,)
-    KWARGS = {"train": False}
-
-    def inject_fake_data(self, tmpdir):
-        create_ncars_folder(tmpdir, datasets.NCARS._TEST_PATH)
-
-
-#############################
-# Gen4AutomotiveDetectionMini
-#############################
-
-
-def create_minidataset_folder(tmpdir, split_folder):
-    testfolder = Path(
-        tmpdir, "Gen4AutomotiveDetectionMini", "mini_dataset", split_folder
-    )
-    testfolder.mkdir(parents=True, exist_ok=True)
-    shutil.copyfile(
-        PATH_TO_TEST_DATA / "sample_ncars.dat",
-        Path(testfolder, "moorea_2019-02-19_004_td_610500000_670500000_td.dat"),
-    )
-    label = np.array([0.4, 0.2, 0.1, 0.1])
-    np.save(testfolder / "moorea_2019-02-19_004_td_610500000_670500000_bbox.npy", label)
-
-
-class Gen4AutomotiveDetectionMiniTestCase_Train(dataset_utils.DatasetTestCase):
-    DATASET_CLASS = datasets.Gen4AutomotiveDetectionMini
-    FEATURE_TYPES = (datasets.Gen4AutomotiveDetectionMini._DTYPE,)
-    TARGET_TYPES = (np.array,)
-    KWARGS = {"split": "train"}
-
-    def inject_fake_data(self, tmpdir):
-        create_minidataset_folder(
-            tmpdir, datasets.Gen4AutomotiveDetectionMini._TRAIN_FOLDER
-        )
-
-
-class Gen4AutomotiveDetectionMiniTestCase_Valid(dataset_utils.DatasetTestCase):
-    DATASET_CLASS = datasets.Gen4AutomotiveDetectionMini
-    FEATURE_TYPES = (datasets.Gen4AutomotiveDetectionMini._DTYPE,)
-    TARGET_TYPES = (np.array,)
-    KWARGS = {"split": "valid"}
-
-    def inject_fake_data(self, tmpdir):
-        create_minidataset_folder(
-            tmpdir, datasets.Gen4AutomotiveDetectionMini._VALID_FOLDER
-        )
-
-
-class Gen4AutomotiveDetectionMiniTestCase_Test(dataset_utils.DatasetTestCase):
-    DATASET_CLASS = datasets.Gen4AutomotiveDetectionMini
-    FEATURE_TYPES = (datasets.Gen4AutomotiveDetectionMini._DTYPE,)
-    TARGET_TYPES = (np.array,)
-    KWARGS = {"split": "test"}
-
-    def inject_fake_data(self, tmpdir):
-        create_minidataset_folder(
-            tmpdir, datasets.Gen4AutomotiveDetectionMini._TEST_FOLDER
-        )
-
-
-#########################
-# Gen1AutomotiveDetection
-#########################
-
-
-def create_gen1dataset_folder(tmpdir, split_folder):
-    testfolder = Path(tmpdir, "detection_dataset_duration_60s_ratio_1.0", split_folder)
-    testfolder.mkdir(parents=True, exist_ok=True)
-    shutil.copyfile(
-        PATH_TO_TEST_DATA / "sample_ncars.dat",
-        Path(testfolder, "moorea_2019-02-19_004_td_610500000_670500000_td.dat"),
-    )
-    label = np.array([0.4, 0.2, 0.1, 0.1])
-    np.save(testfolder / "moorea_2019-02-19_004_td_610500000_670500000_bbox.npy", label)
-
-
-class Gen1AutomotiveDetectionTestCase_Train(dataset_utils.DatasetTestCase):
-    DATASET_CLASS = datasets.Gen1AutomotiveDetection
-    FEATURE_TYPES = (datasets.Gen1AutomotiveDetection._DTYPE,)
-    TARGET_TYPES = (np.array,)
-    KWARGS = {"split": "train"}
-
-    def inject_fake_data(self, tmpdir):
-        create_gen1dataset_folder(
-            tmpdir, datasets.Gen1AutomotiveDetection._TRAIN_FOLDER
-        )
-
-
-class Gen1AutomotiveDetectionTestCase_Valid(dataset_utils.DatasetTestCase):
-    DATASET_CLASS = datasets.Gen1AutomotiveDetection
-    FEATURE_TYPES = (datasets.Gen1AutomotiveDetection._DTYPE,)
-    TARGET_TYPES = (np.array,)
-    KWARGS = {"split": "valid"}
-
-    def inject_fake_data(self, tmpdir):
-        create_gen1dataset_folder(
-            tmpdir, datasets.Gen1AutomotiveDetection._VALID_FOLDER
-        )
-
-
-class Gen1AutomotiveDetectionTestCase_Test(dataset_utils.DatasetTestCase):
-    DATASET_CLASS = datasets.Gen1AutomotiveDetection
-    FEATURE_TYPES = (datasets.Gen1AutomotiveDetection._DTYPE,)
-    TARGET_TYPES = (np.array,)
-    KWARGS = {"split": "test"}
-
-    def inject_fake_data(self, tmpdir):
-        create_gen1dataset_folder(tmpdir, datasets.Gen1AutomotiveDetection._TEST_FOLDER)

tonic-1.5.1.dev19/tonic/datasets/nerdd.py

@@ -1,120 +0,0 @@
-import os
-import numpy as np
-from typing import Any, Callable, Optional, Tuple
-from tonic.download_utils import extract_archive
-from tonic.io import make_structured_array
-
-class NERDD:
-    """`NeRDD <https://github.com/MagriniGabriele/NeRDD>`_
-    Neuromorphic Event-based Reactive Driving Dataset.
-
-    Parameters:
-        root (string): Location to save files to on disk.
-        transforms (callable, optional): A callable of transforms that is applied to both data and
-                                         labels at the same time.
-    """
-
-    filename = "NeRDD.zip"
-    folder_name = "DATA"
-    sensor_size = (1280, 720, 2)
-    dtype = np.dtype([("t", np.int32), ("x", np.int16), ("y", np.int16), ("p", bool)])
-    ordering = dtype.names
-
-    def __init__(
-        self,
-        root: str,
-        transforms: Optional[Callable] = None,
-    ):
-        self.data = []
-        self.targets = []
-        self.location_on_system = root
-        self.transforms = transforms
-
-        if not self._check_extracted():
-            if not self._check_zip_exists():
-                raise FileNotFoundError(
-                    f"The dataset archive '{self.filename}' was not found in {self.location_on_system}. "
-                    "Please download it manually and place it in the specified directory."
-                )
-            self._extract_archive()
-        
-        self._load_dataset_structure()
-
-    def _check_zip_exists(self) -> bool:
-        """Check if the zip file exists in the specified directory."""
-        return os.path.isfile(os.path.join(self.location_on_system, self.filename))
-
-    def _check_extracted(self) -> bool:
-        """Check if the dataset has been extracted and contains at least 10 .npz files."""
-        data_path = os.path.join(self.location_on_system, self.folder_name)
-        if not os.path.isdir(data_path):
-            return False
-        npz_files = []
-        for root, _, files in os.walk(data_path):
-            npz_files.extend([f for f in files if f.endswith(".npz")])
-        return len(npz_files) == 115
-
-    def _extract_archive(self):
-        """Extract the dataset archive."""
-        print(f"Extracting {self.filename}...")
-        archive_path = os.path.join(self.location_on_system, self.filename)
-        extract_archive(archive_path)
-        print(f"Extraction complete. Files are now in {self.location_on_system}.")
-
-    def _load_dataset_structure(self):
-        """Load the dataset files and their corresponding labels."""
-        data_path = os.path.join(self.location_on_system, self.folder_name)
-        archives = [
-            f for f in os.listdir(data_path) 
-            if os.path.isdir(os.path.join(data_path, f))
-        ]
-        
-        for archive in archives:
-            archive_path = os.path.join(data_path, archive)
-            scenes = [
-                f for f in os.listdir(archive_path) 
-                if os.path.isdir(os.path.join(archive_path, f))
-            ]
-            
-            for scene in scenes:
-                scene_path = os.path.join(archive_path, scene)
-                event_file = os.path.join(scene_path, "Event", "output_events.npz")
-                rgb_folder = os.path.join(scene_path, "rgb_folder")
-                label_file = os.path.join(scene_path, "ev_rgb_coordinates.txt")
-                
-                if os.path.exists(event_file) and os.path.exists(label_file):
-                    self.data.append((event_file, label_file))
-                else:
-                    print(f"Skipping scene {scene} in archive {archive}: missing files.")
-
-    def __getitem__(self, index: int) -> Tuple[Any, Any]:
-        """
-        Returns:
-            (events, target) where target is index of the target class.
-        """
-        event_file, label_file = self.data[index]
-        events = np.load(event_file)["data"]
-        events = make_structured_array(
-            events[:, 3],  # t
-            events[:, 0],  # x
-            events[:, 1],  # y
-            events[:, 2],  # p
-            dtype=self.dtype,
-        )
-
-        with open(label_file, "r") as f:
-            bboxes = []
-            for line in f.readlines():
-                timestamp, bbox_str = line.strip().split(":")
-                bbox_coords = list(map(float, bbox_str.split(",")))
-                bboxes.append([float(timestamp)] + bbox_coords)
-            bboxes = np.array(bboxes, dtype=np.float32)
-
-        # Apply transforms if provided
-        if self.transforms is not None:
-            events, bboxes = self.transforms(events, bboxes)
-
-        return events, bboxes
-
-    def __len__(self):
-        return len(self.data)