mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-24 10:16:09 +00:00
hotfix: flip actions
This commit is contained in:
@@ -311,6 +311,7 @@ class LiberoEnv(gym.Env):
|
|||||||
|
|
||||||
def step(self, action):
|
def step(self, action):
|
||||||
assert action.ndim == 1
|
assert action.ndim == 1
|
||||||
|
action[-1] = 1.0 - action[-1]
|
||||||
raw_obs, reward, done, info = self._env.step(action)
|
raw_obs, reward, done, info = self._env.step(action)
|
||||||
is_success = self._env.check_success()
|
is_success = self._env.check_success()
|
||||||
terminated = done or is_success
|
terminated = done or is_success
|
||||||
|
|||||||
Reference in New Issue
Block a user