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

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,

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,