Revert ignore comments.

argoverse · Dec 11, 2023 · 9f350ca · 9f350ca
1 parent 85ae883
commit 9f350ca
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/av2/torch/data_loaders/detection.py b/src/av2/torch/data_loaders/detection.py
@@ -19,7 +19,7 @@
 
 
 @dataclass
-class DetectionDataLoader(Dataset[Sweep]):
+class DetectionDataLoader(Dataset[Sweep]):  # type: ignore
     """PyTorch data-loader for the sensor dataset.
 
     The sensor dataset should exist somewhere such as `~/data/datasets/{dataset_name}/{dataset_type}/{split_name}`,

diff --git a/src/av2/torch/data_loaders/scene_flow.py b/src/av2/torch/data_loaders/scene_flow.py
@@ -23,7 +23,7 @@
 
 
 @dataclass
-class SceneFlowDataloader(Dataset[Tuple[Sweep, Sweep, Se3, Optional[Flow]]]):
+class SceneFlowDataloader(Dataset[Tuple[Sweep, Sweep, Se3, Optional[Flow]]]):  # type: ignore
     """PyTorch data-loader for the sensor dataset.
 
     Args: