revert changes to save_policy_to_safetensors

2024-05-16 14:50:47 +01:00 · 2024-05-16 14:50:47 +01:00 · d0739e1c60
parent 92cd9d33f1
commit d0739e1c60
3 changed files with 1 additions and 1 deletions
--- a/tests/data/save_policy_to_safetensors/aloha_act/actions.safetensors
+++ b/tests/data/save_policy_to_safetensors/aloha_act/actions.safetensors
--- a/tests/data/save_policy_to_safetensors/aloha_act/grad_stats.safetensors
+++ b/tests/data/save_policy_to_safetensors/aloha_act/grad_stats.safetensors
--- a/tests/scripts/save_policy_to_safetensor.py
+++ b/tests/scripts/save_policy_to_safetensor.py
@ -105,6 +105,6 @@ def save_policy_to_safetensors(output_dir, env_name, policy_name, extra_override
 if __name__ == "__main__":
    # Instructions: include the policies that you want to save artifacts for here. Please make sure to revert
    # your changes when you are done.
-    env_policies = [("aloha", "act", ["policy.n_action_steps=10"])]
+    env_policies = []
    for env, policy, extra_overrides in env_policies:
        save_policy_to_safetensors("tests/data/save_policy_to_safetensors", env, policy, extra_overrides)