v2.0.22: Fixed typo log subset

kylevedder · Aug 16, 2024 · 5e8604c · 5e8604c
1 parent 2d8a6d2
commit 5e8604c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/bucketed_scene_flow_eval/datasets/waymoopen/dataset.py b/bucketed_scene_flow_eval/datasets/waymoopen/dataset.py
@@ -50,7 +50,9 @@ def __init__(
         log_subset: list[str] | None = None,
         eval_args=dict(),
     ) -> None:
-        self.sequence_loader = WaymoSupervisedSceneFlowSequenceLoader(root_dir)
+        self.sequence_loader = WaymoSupervisedSceneFlowSequenceLoader(
+            root_dir, log_subset=log_subset
+        )
         super().__init__(
             sequence_loader=self.sequence_loader,
             subsequence_length=subsequence_length,
@@ -59,7 +61,6 @@ def __init__(
             eval_type=eval_type,
             eval_args=eval_args,
             use_cache=use_cache,
-            log_subset=log_subset,
         )
 
     def evaluator(self) -> Evaluator:
@@ -78,7 +79,9 @@ def __init__(
         log_subset: list[str] | None = None,
         eval_args=dict(),
     ) -> None:
-        self.sequence_loader = WaymoSupervisedSceneFlowSequenceLoader(root_dir)
+        self.sequence_loader = WaymoSupervisedSceneFlowSequenceLoader(
+            root_dir, log_subset=log_subset
+        )
         super().__init__(
             sequence_loader=self.sequence_loader,
             subsequence_length=subsequence_length,
@@ -87,7 +90,6 @@ def __init__(
             eval_type=eval_type,
             eval_args=eval_args,
             use_cache=use_cache,
-            log_subset=log_subset,
         )
 
     def evaluator(self) -> Evaluator:

diff --git a/pyproject.toml b/pyproject.toml
@@ -15,7 +15,7 @@ exclude = [
 
 [project]
 name = "bucketed_scene_flow_eval"
-version = "2.0.21"
+version = "2.0.22"
 authors = [
   { name="Kyle Vedder", email="[email protected]" },
 ]