admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-06-29 14:17:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixing stale assertion

Browse Source
This commit is contained in:
Maxime Ellerbach
2026-06-22 13:42:21 +00:00
parent 84d33bf066
commit 5f6ddab629
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/fastwam/test_fastwam_policy.py
+3 -2
View File
@@ -77,7 +77,7 @@ def test_config_validates_features_model_ids_and_saved_auto_route(tmp_path):
FastWAMConfig(tokenizer_model_id="somebody/other-tokenizer")
def test_preprocessor_normalizes_images_and_postprocessor_toggles_actions(tmp_path):
def test_preprocessor_passes_images_through_and_postprocessor_toggles_actions(tmp_path):
cfg = FastWAMConfig(
action_dim=3,
proprio_dim=2,
@@ -127,8 +127,9 @@ def test_preprocessor_normalizes_images_and_postprocessor_toggles_actions(tmp_pa
postprocessor.save_pretrained(tmp_path, config_filename="policy_postprocessor.json")
_, loaded_postprocessor = make_pre_post_processors(cfg, pretrained_path=str(tmp_path))
# VISUAL normalization is IDENTITY
expected_image = torch.tensor(
[[[[-1.0, 0.0], [1.0, 0.0]], [[-1.0, 0.0], [1.0, 0.0]], [[-1.0, 0.0], [1.0, 0.0]]]]
[[[[0.0, 0.5], [1.0, 0.5]], [[0.0, 0.5], [1.0, 0.5]], [[0.0, 0.5], [1.0, 0.5]]]]
)
assert preprocessor.name == "policy_preprocessor"
assert postprocessor.name == "policy_postprocessor"