pass step as kwarg

2024-04-15 09:52:54 +01:00
parent 2ccf89d78c
commit 9241b5e830
2 changed files with 2 additions and 2 deletions
--- a/lerobot/scripts/eval.py
+++ b/lerobot/scripts/eval.py
@@ -130,7 +130,7 @@ def eval_policy(

        # get the next action for the environment
        with torch.inference_mode():
-            action = policy.select_action(observation, step)
+            action = policy.select_action(observation, step=step)

        # apply inverse transform to unnormalize the action
        action = postprocess_action(action, transform)