From cf36f4b8732011d40bd8fab605b390a588d77567 Mon Sep 17 00:00:00 2001 From: Pepijn Date: Tue, 2 Dec 2025 17:26:44 +0100 Subject: [PATCH] add localdir --- examples/dataset/slurm_port_egodex.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) 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)