admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-20 19:19:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(eval): specify type parameters for preprocessor and postprocessor in eval_policy function (#1904)

Browse Source
This commit is contained in:
Adil Zouitine
2025-09-10 10:31:05 +02:00
committed by GitHub
parent e881fb6678
commit 7e30090e97
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/scripts/eval.py
+2 -2
View File
@@ -220,8 +220,8 @@ def rollout(
def eval_policy( def eval_policy(
env: gym.vector.VectorEnv, env: gym.vector.VectorEnv,
policy: PreTrainedPolicy, policy: PreTrainedPolicy,
preprocessor: PolicyProcessorPipeline, preprocessor: PolicyProcessorPipeline[dict[str, Any]],
postprocessor: PolicyProcessorPipeline, postprocessor: PolicyProcessorPipeline[dict[str, Any]],
n_episodes: int, n_episodes: int,
max_episodes_rendered: int = 0, max_episodes_rendered: int = 0,
videos_dir: Path | None = None, videos_dir: Path | None = None,