feat: Add root directory option for dataset configuration (#765)

Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
This commit is contained in:
Haskely
2025-02-26 00:27:36 +08:00
committed by GitHub
parent 3354d919fc
commit eecf32e77a
2 changed files with 6 additions and 1 deletions

View File

@@ -29,6 +29,8 @@ class DatasetConfig:
# "dataset_index" into the returned item. The index mapping is made according to the order in which the
# datsets are provided.
repo_id: str
# Root directory where the dataset will be stored (e.g. 'dataset/path').
root: str | None = None
episodes: list[int] | None = None
image_transforms: ImageTransformsConfig = field(default_factory=ImageTransformsConfig)
revision: str | None = None