diff --git a/lerobot/scripts/merge.py b/lerobot/scripts/merge.py index 61bab7f8..8cf0dffd 100644 --- a/lerobot/scripts/merge.py +++ b/lerobot/scripts/merge.py @@ -1293,20 +1293,16 @@ def merge_datasets( if __name__ == "__main__": # Set up argument parser - parser = argparse.ArgumentParser(description='Merge datasets from multiple sources.') - + parser = argparse.ArgumentParser(description="Merge datasets from multiple sources.") + # Add arguments - parser.add_argument('--sources', nargs='+', required=True, - help='List of source folder paths') - parser.add_argument('--output', required=True, - help='Output folder path') - parser.add_argument('--max_dim', type=int, default=32, - help='Maximum dimension (default: 32)') - parser.add_argument('--fps', type=int, default=20, - help='Your datasets FPS (default: 20)') - + parser.add_argument("--sources", nargs="+", required=True, help="List of source folder paths") + parser.add_argument("--output", required=True, help="Output folder path") + parser.add_argument("--max_dim", type=int, default=32, help="Maximum dimension (default: 32)") + parser.add_argument("--fps", type=int, default=20, help="Your datasets FPS (default: 20)") + # Parse arguments args = parser.parse_args() - + # Use parsed arguments merge_datasets(args.sources, args.output, max_dim=args.max_dim, default_fps=args.fps)