wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

nit

This commit is contained in:
Cadene 2024-04-24 15:56:22 +00:00
parent 72751b7cf6
commit 343579368b
1 changed files with 0 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/test_policies.py
View File

@ -194,10 +194,3 @@ def test_normalize(insert_temporal_dim):
new_unnormalize = Unnormalize(output_shapes, unnormalize_output_modes, stats=None) new_unnormalize = Unnormalize(output_shapes, unnormalize_output_modes, stats=None)
new_unnormalize.load_state_dict(unnormalize.state_dict()) new_unnormalize.load_state_dict(unnormalize.state_dict())
unnormalize(output_batch) unnormalize(output_batch)
if __name__ == "__main__":
test_policy(
*("aloha", "act", ["env.task=AlohaTransferCube-v0", "dataset_id=aloha_sim_transfer_cube_scripted"])
)
# test_policy(insert_temporal_dim=True)