From de5c30405e55bc37365b9700f66a42b1a180ccef Mon Sep 17 00:00:00 2001 From: Simon Alibert Date: Mon, 25 Mar 2024 12:35:06 +0100 Subject: [PATCH] fix wrong version --- lerobot/common/datasets/abstract.py | 2 +- lerobot/common/datasets/simxarm.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/lerobot/common/datasets/abstract.py b/lerobot/common/datasets/abstract.py index 8d17cd78..7b8a8423 100644 --- a/lerobot/common/datasets/abstract.py +++ b/lerobot/common/datasets/abstract.py @@ -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, diff --git a/lerobot/common/datasets/simxarm.py b/lerobot/common/datasets/simxarm.py index af006f76..3b1c75be 100644 --- a/lerobot/common/datasets/simxarm.py +++ b/lerobot/common/datasets/simxarm.py @@ -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,