Add regression tests (#119)

Browse Source

- Add `tests/scripts/save_policy_to_safetensor.py` to generate test artifacts
- Add `test_backward_compatibility to test generated outputs from the policies against artifacts

...

This commit is contained in:

Simon Alibert

2024-05-04 16:20:30 +02:00

committed by GitHub

parent 19812ca470

commit c77633c38c

15 changed files with 236 additions and 43 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

BIN

tests/data/save_policy_to_safetensors/aloha_act/actions.safetensors Normal file

Binary file not shown.

BIN

tests/data/save_policy_to_safetensors/aloha_act/grad_stats.safetensors Normal file

Binary file not shown.

BIN

tests/data/save_policy_to_safetensors/aloha_act/output_dict.safetensors Normal file

Binary file not shown.

BIN

tests/data/save_policy_to_safetensors/aloha_act/param_stats.safetensors Normal file

Binary file not shown.

BIN

tests/data/save_policy_to_safetensors/pusht_diffusion/actions.safetensors Normal file

Binary file not shown.

BIN

tests/data/save_policy_to_safetensors/pusht_diffusion/grad_stats.safetensors Normal file

Binary file not shown.

BIN

tests/data/save_policy_to_safetensors/pusht_diffusion/output_dict.safetensors Normal file

Binary file not shown.

BIN

tests/data/save_policy_to_safetensors/pusht_diffusion/param_stats.safetensors Normal file

Binary file not shown.