wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix wrong version

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
Show Stats 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__( def __init__(
self, self,
dataset_id: str, dataset_id: str,
version: str | None = "v1.0", version: str | None,
batch_size: int = None, batch_size: int = None,
*, *,
shuffle: bool = True, shuffle: bool = True,

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

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