diff --git a/tests/scripts/save_policy_to_safetensor.py b/tests/scripts/save_policy_to_safetensor.py index 06763d72..70337c17 100644 --- a/tests/scripts/save_policy_to_safetensor.py +++ b/tests/scripts/save_policy_to_safetensor.py @@ -89,7 +89,7 @@ def save_policy_to_safetensors(output_dir, env_name, policy_name, extra_override if __name__ == "__main__": env_policies = [ - ("xarm", "tdmpc", ["policy.n_action_repeats=2"]), + # ("xarm", "tdmpc", ["policy.n_action_repeats=2"]), ( "pusht", "diffusion", diff --git a/tests/test_policies.py b/tests/test_policies.py index efc33281..d7b7cb50 100644 --- a/tests/test_policies.py +++ b/tests/test_policies.py @@ -239,7 +239,7 @@ def test_normalize(insert_temporal_dim): @pytest.mark.parametrize( "env_name, policy_name, extra_overrides", [ - ("xarm", "tdmpc", ["policy.n_action_repeats=2"]), + # ("xarm", "tdmpc", ["policy.n_action_repeats=2"]), ( "pusht", "diffusion",