diff --git a/tests/data/save_policy_to_safetensors/dora_aloha_real_act_real/actions.safetensors b/tests/data/save_policy_to_safetensors/dora_aloha_real_act_aloha_real/actions.safetensors similarity index 100% rename from tests/data/save_policy_to_safetensors/dora_aloha_real_act_real/actions.safetensors rename to tests/data/save_policy_to_safetensors/dora_aloha_real_act_aloha_real/actions.safetensors diff --git a/tests/data/save_policy_to_safetensors/dora_aloha_real_act_real/grad_stats.safetensors b/tests/data/save_policy_to_safetensors/dora_aloha_real_act_aloha_real/grad_stats.safetensors similarity index 100% rename from tests/data/save_policy_to_safetensors/dora_aloha_real_act_real/grad_stats.safetensors rename to tests/data/save_policy_to_safetensors/dora_aloha_real_act_aloha_real/grad_stats.safetensors diff --git a/tests/data/save_policy_to_safetensors/dora_aloha_real_act_real/output_dict.safetensors b/tests/data/save_policy_to_safetensors/dora_aloha_real_act_aloha_real/output_dict.safetensors similarity index 100% rename from tests/data/save_policy_to_safetensors/dora_aloha_real_act_real/output_dict.safetensors rename to tests/data/save_policy_to_safetensors/dora_aloha_real_act_aloha_real/output_dict.safetensors diff --git a/tests/data/save_policy_to_safetensors/dora_aloha_real_act_real/param_stats.safetensors b/tests/data/save_policy_to_safetensors/dora_aloha_real_act_aloha_real/param_stats.safetensors similarity index 100% rename from tests/data/save_policy_to_safetensors/dora_aloha_real_act_real/param_stats.safetensors rename to tests/data/save_policy_to_safetensors/dora_aloha_real_act_aloha_real/param_stats.safetensors diff --git a/tests/data/save_policy_to_safetensors/dora_aloha_real_act_real_no_state/actions.safetensors b/tests/data/save_policy_to_safetensors/dora_aloha_real_act_real_no_state/actions.safetensors deleted file mode 100644 index 2e26ef270..000000000 --- a/tests/data/save_policy_to_safetensors/dora_aloha_real_act_real_no_state/actions.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:b5a9f73a2356aff9c717cdfd0d37a6da08b0cf2cc09c98edbc9492501b7f64a5 -size 5104 diff --git a/tests/data/save_policy_to_safetensors/dora_aloha_real_act_real_no_state/grad_stats.safetensors b/tests/data/save_policy_to_safetensors/dora_aloha_real_act_real_no_state/grad_stats.safetensors deleted file mode 100644 index b959bc6e0..000000000 --- a/tests/data/save_policy_to_safetensors/dora_aloha_real_act_real_no_state/grad_stats.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:28738b3cfad17af0ac5181effdd796acdf7953cd5bcca3f421a11ddfd6b0076f -size 30800 diff --git a/tests/data/save_policy_to_safetensors/dora_aloha_real_act_real_no_state/output_dict.safetensors b/tests/data/save_policy_to_safetensors/dora_aloha_real_act_real_no_state/output_dict.safetensors deleted file mode 100644 index 455834aa8..000000000 --- a/tests/data/save_policy_to_safetensors/dora_aloha_real_act_real_no_state/output_dict.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:4bb8a197a40456fdbc16029126268e6bcef3eca1837d88235165dc7e14618bea -size 68 diff --git a/tests/data/save_policy_to_safetensors/dora_aloha_real_act_real_no_state/param_stats.safetensors b/tests/data/save_policy_to_safetensors/dora_aloha_real_act_real_no_state/param_stats.safetensors deleted file mode 100644 index d50fb31d9..000000000 --- a/tests/data/save_policy_to_safetensors/dora_aloha_real_act_real_no_state/param_stats.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:bea60cce42d324f539dd3bca1e66b5ba6391838fdcadb00efc25f3240edb529a -size 33600 diff --git a/tests/test_policies.py b/tests/test_policies.py index d90f00716..f358170d1 100644 --- a/tests/test_policies.py +++ b/tests/test_policies.py @@ -367,8 +367,7 @@ def test_normalize(insert_temporal_dim): ), ("aloha", "act", ["policy.n_action_steps=10"], ""), ("aloha", "act", ["policy.n_action_steps=1000", "policy.chunk_size=1000"], "_1000_steps"), - ("dora_aloha_real", "act_real", ["policy.n_action_steps=10"], ""), - ("dora_aloha_real", "act_real_no_state", ["policy.n_action_steps=10"], ""), + ("dora_aloha_real", "act_aloha_real", ["policy.n_action_steps=10"], ""), ], ) # As artifacts have been generated on an x86_64 kernel, this test won't