diff --git a/examples/dataset/slurm_port_egodex.py b/examples/dataset/slurm_port_egodex.py index 71f962416..ec9c73e26 100644 --- a/examples/dataset/slurm_port_egodex.py +++ b/examples/dataset/slurm_port_egodex.py @@ -67,7 +67,7 @@ class PortEgoDexShards(PipelineStep): def _get_state_for_single_frame(transforms_group, frame_idx): """ - Construct 48D hand state representation from EgoDex paper. + Construct 48D hand state representation from EgoDex. State vector composition (per hand = 24D, total = 48D): - Wrist 3D position (3) @@ -421,6 +421,12 @@ def main(): default=100.0, help="Percentage of dataset to process (e.g., 1.0 for 1%%). Useful for testing.", ) + parser.add_argument( + "--local-dir", + type=Path, + default=None, + help="Local directory to save the LeRobot dataset. Defaults to data/local_datasets.", + ) args = parser.parse_args() kwargs = vars(args)