hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

fix wrong version

Browse Source
This commit is contained in:
Simon Alibert
2024-03-25 12:35:06 +01:00
parent 462e7469e8
commit de5c30405e
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/common/datasets/abstract.py
View File

@@ -21,7 +21,7 @@ class AbstractExperienceReplay(TensorDictReplayBuffer):
def __init__(
self,
dataset_id: str,
version: str | None = "v1.0",
version: str | None,
batch_size: int = None,
*,
shuffle: bool = True,

2
lerobot/common/datasets/simxarm.py
View File

@@ -40,7 +40,7 @@ class SimxarmExperienceReplay(AbstractExperienceReplay):
def __init__(
self,
dataset_id: str,
version: str | None = None,
version: str | None = "v1.0",
batch_size: int = None,
*,
shuffle: bool = True,