From 943295eba6940a29d81202d2447bd443acce0afc Mon Sep 17 00:00:00 2001 From: Maximellerbach Date: Tue, 28 Apr 2026 16:50:31 +0200 Subject: [PATCH] use consistantly int for data and video_files_size_in_mb --- src/lerobot/datasets/aggregate.py | 4 ++-- tests/fixtures/dataset_factories.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/lerobot/datasets/aggregate.py b/src/lerobot/datasets/aggregate.py index cac009b64..90fc8f583 100644 --- a/src/lerobot/datasets/aggregate.py +++ b/src/lerobot/datasets/aggregate.py @@ -237,8 +237,8 @@ def aggregate_datasets( aggr_repo_id: str, roots: list[Path] | None = None, aggr_root: Path | None = None, - data_files_size_in_mb: float | None = None, - video_files_size_in_mb: float | None = None, + data_files_size_in_mb: int | None = None, + video_files_size_in_mb: int | None = None, chunk_size: int | None = None, ): """Aggregates multiple LeRobot datasets into a single unified dataset. diff --git a/tests/fixtures/dataset_factories.py b/tests/fixtures/dataset_factories.py index 138eb06a8..48128a8d0 100644 --- a/tests/fixtures/dataset_factories.py +++ b/tests/fixtures/dataset_factories.py @@ -158,8 +158,8 @@ def info_factory(features_factory): total_frames: int = 0, total_tasks: int = 0, chunks_size: int = DEFAULT_CHUNK_SIZE, - data_files_size_in_mb: float = DEFAULT_DATA_FILE_SIZE_IN_MB, - video_files_size_in_mb: float = DEFAULT_VIDEO_FILE_SIZE_IN_MB, + data_files_size_in_mb: int = DEFAULT_DATA_FILE_SIZE_IN_MB, + video_files_size_in_mb: int = DEFAULT_VIDEO_FILE_SIZE_IN_MB, data_path: str = DEFAULT_DATA_PATH, video_path: str = DEFAULT_VIDEO_PATH, motor_features: dict = DUMMY_MOTOR_FEATURES,