From e47856add6edcdceed6142373486430f9d2ab311 Mon Sep 17 00:00:00 2001 From: Remi Cadene Date: Sun, 15 Sep 2024 17:22:12 +0200 Subject: [PATCH] Fix unit test test_policies, backward, Remove no_state from test --- .../actions.safetensors | 0 .../grad_stats.safetensors | 0 .../output_dict.safetensors | 0 .../param_stats.safetensors | 0 .../dora_aloha_real_act_real_no_state/actions.safetensors | 3 --- .../dora_aloha_real_act_real_no_state/grad_stats.safetensors | 3 --- .../dora_aloha_real_act_real_no_state/output_dict.safetensors | 3 --- .../dora_aloha_real_act_real_no_state/param_stats.safetensors | 3 --- tests/test_policies.py | 3 +-- 9 files changed, 1 insertion(+), 14 deletions(-) rename tests/data/save_policy_to_safetensors/{dora_aloha_real_act_real => dora_aloha_real_act_aloha_real}/actions.safetensors (100%) rename tests/data/save_policy_to_safetensors/{dora_aloha_real_act_real => dora_aloha_real_act_aloha_real}/grad_stats.safetensors (100%) rename tests/data/save_policy_to_safetensors/{dora_aloha_real_act_real => dora_aloha_real_act_aloha_real}/output_dict.safetensors (100%) rename tests/data/save_policy_to_safetensors/{dora_aloha_real_act_real => dora_aloha_real_act_aloha_real}/param_stats.safetensors (100%) delete mode 100644 tests/data/save_policy_to_safetensors/dora_aloha_real_act_real_no_state/actions.safetensors delete mode 100644 tests/data/save_policy_to_safetensors/dora_aloha_real_act_real_no_state/grad_stats.safetensors delete mode 100644 tests/data/save_policy_to_safetensors/dora_aloha_real_act_real_no_state/output_dict.safetensors delete mode 100644 tests/data/save_policy_to_safetensors/dora_aloha_real_act_real_no_state/param_stats.safetensors 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