Added basic tests for box loading

bucketed_scene_flow_eval/datasets/argoverse2/argoverse_scene_flow.py

@@ -264,6 +264,7 @@ def _subset_log(self, log_subset: Optional[list[str]]):
             self.sequence_id_lst = [
                 sequence_id for sequence_id in self.sequence_id_lst if sequence_id in log_subset
             ]
+            assert len(self.sequence_id_lst) > 0, f"No sequences found in log_subset {log_subset}"
 
     def _sanitize_raw_data_path(self, raw_data_path: Union[Path, list[Path]]) -> list[Path]:
         if isinstance(raw_data_path, str):

bucketed_scene_flow_eval/datasets/argoverse2/dataset.py

@@ -70,11 +70,17 @@ def __init__(
         flow_data_path: Optional[Union[Path, list[Path]]] = None,
         eval_type: str = "bucketed_epe",
         eval_args=dict(),
+        load_boxes: bool = False,
         load_flow: bool = True,
         use_cache=True,
         **kwargs,
     ) -> None:
-        if load_flow:
+        if load_boxes:
+            self.sequence_loader = ArgoverseBoxAnnotationSequenceLoader(
+                root_dir,
+                **kwargs,
+            )
+        elif load_flow:
             self.sequence_loader = ArgoverseSceneFlowSequenceLoader(
                 root_dir,
                 use_gt_flow=use_gt_flow,