diff --git a/src/lerobot/scripts/eval.py b/src/lerobot/scripts/eval.py index 87058abb8..38501a2e2 100644 --- a/src/lerobot/scripts/eval.py +++ b/src/lerobot/scripts/eval.py @@ -220,8 +220,8 @@ def rollout( def eval_policy( env: gym.vector.VectorEnv, policy: PreTrainedPolicy, - preprocessor: PolicyProcessorPipeline, - postprocessor: PolicyProcessorPipeline, + preprocessor: PolicyProcessorPipeline[dict[str, Any]], + postprocessor: PolicyProcessorPipeline[dict[str, Any]], n_episodes: int, max_episodes_rendered: int = 0, videos_dir: Path | None = None,