Fixed bug in the action scale of the intervention actions and offline dataset actions. (scale by inverse delta)

Co-authored-by: Adil Zouitine <adizouitinegm@gmail.com>
This commit is contained in:
Michel Aractingi
2025-02-14 15:17:16 +01:00
parent 8209a6dfb7
commit a16ea283f5
5 changed files with 42 additions and 16 deletions

View File

@@ -209,6 +209,7 @@ class ReplayBuffer:
state_keys: Optional[Sequence[str]] = None,
capacity: Optional[int] = None,
action_mask: Optional[Sequence[int]] = None,
action_delta: Optional[float] = None,
) -> "ReplayBuffer":
"""
Convert a LeRobotDataset into a ReplayBuffer.
@@ -249,6 +250,9 @@ class ReplayBuffer:
else:
data["action"] = data["action"][:, action_mask]
if action_delta is not None:
data["action"] = data["action"] / action_delta
replay_buffer.add(
state=data["state"],
action=data["action"],