ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-06-04 21:01:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(processors): several additions (#1926)

Browse Source 
* chore(processor): remove merge_transitions functions (#1925)

* refactor(processors): move processors out of configs (#1927)

* chore(processor): streamline combine_features_dict (#1928)

* chore(policies): use new constants (#1929)

* fix(deps): right version transformers (#1930)

* fix(tests): add none + disable async tests for now (#1931)
This commit is contained in:
Steven Palma
2025-09-13 23:53:20 +02:00
committed by GitHub
parent 839ac5f2aa
commit 50293bb17b
18 changed files with 154 additions and 400 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/lerobot/scripts/eval.py
View File

@@ -158,8 +158,8 @@ def rollout(
observation = add_envs_task(env, observation)
observation = preprocessor(observation)
with torch.inference_mode():
action: PolicyAction = policy.select_action(observation)
action: PolicyAction = postprocessor(action)
action = policy.select_action(observation)
action = postprocessor(action)
# Convert to CPU / numpy.
action_numpy: np.ndarray = action.to("cpu").numpy()