From 4bde95f93a13202bd11d3a48c7db64900f6581f6 Mon Sep 17 00:00:00 2001
From: Simon Alibert <alibert.sim@gmail.com>
Date: Fri, 3 May 2024 11:08:04 +0200
Subject: [PATCH] Disable test on tdmpc

---
 tests/scripts/save_policy_to_safetensor.py | 2 +-
 tests/test_policies.py                     | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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",