From 7e30090e973b099a4cd6b60c3b4d8eaaee00b440 Mon Sep 17 00:00:00 2001
From: Adil Zouitine <adilzouitinegm@gmail.com>
Date: Wed, 10 Sep 2025 10:31:05 +0200
Subject: [PATCH] refactor(eval): specify type parameters for preprocessor and
 postprocessor in eval_policy function (#1904)

---
 src/lerobot/scripts/eval.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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,