From 391355c9913d8239e6f317e1656954d2dcccec91 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 10 Apr 2025 04:23:50 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- lerobot/scripts/merge.py | 20 ++++++++------------ 1 file changed, 8 insertions(+), 12 deletions(-) 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)