Enable tests for TD-MPC (#160)

2024-05-09 13:42:12 +01:00
parent 7bb5b15f4c
commit e89521dfa0
7 changed files with 5 additions and 6 deletions
--- a/tests/data/save_policy_to_safetensors/xarm_tdmpc/actions.safetensors
+++ b/tests/data/save_policy_to_safetensors/xarm_tdmpc/actions.safetensors
--- a/tests/data/save_policy_to_safetensors/xarm_tdmpc/grad_stats.safetensors
+++ b/tests/data/save_policy_to_safetensors/xarm_tdmpc/grad_stats.safetensors
--- a/tests/data/save_policy_to_safetensors/xarm_tdmpc/output_dict.safetensors
+++ b/tests/data/save_policy_to_safetensors/xarm_tdmpc/output_dict.safetensors
--- a/tests/data/save_policy_to_safetensors/xarm_tdmpc/param_stats.safetensors
+++ b/tests/data/save_policy_to_safetensors/xarm_tdmpc/param_stats.safetensors
--- a/tests/test_policies.py
+++ b/tests/test_policies.py
@@ -236,7 +236,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", []),
        (
            "pusht",
            "diffusion",