diff --git a/lerobot/scripts/eval.py b/lerobot/scripts/eval.py index d676623e..2b8906d7 100644 --- 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)