Update lerobot/scripts/eval.py

Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
This commit is contained in:
Remi 2024-04-10 18:07:27 +02:00 committed by GitHub
parent 0f0113a7a6
commit 4c3d8b061e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions

View File

@ -135,7 +135,7 @@ def eval_policy(
# apply inverse transform to unnormalize the action # apply inverse transform to unnormalize the action
action = postprocess_action(action, transform) action = postprocess_action(action, transform)
# apply the next # apply the next action
observation, reward, terminated, truncated, info = env.step(action) observation, reward, terminated, truncated, info = env.step(action)
if max_episodes_rendered > 0: if max_episodes_rendered > 0:
render_frame(env) render_frame(env)