test_datasets.py are passing!

2024-04-08 14:02:03 +00:00
parent e1ac5dc62f
commit 70aaf1c4cb
109 changed files with 90 additions and 228 deletions
--- a/tests/test_policies.py
+++ b/tests/test_policies.py
@@ -13,7 +13,7 @@ from .utils import DEVICE, DEFAULT_CONFIG_PATH
    "env_name,policy_name,extra_overrides",
    [
        ("simxarm", "tdmpc", ["policy.mpc=true"]),
-        #("pusht", "tdmpc", ["policy.mpc=false"]),
+        ("pusht", "tdmpc", ["policy.mpc=false"]),
        ("pusht", "diffusion", []),
        # ("aloha", "act", ["env.task=sim_insertion", "dataset_id=aloha_sim_insertion_human"]),
        #("aloha", "act", ["env.task=sim_insertion", "dataset_id=aloha_sim_insertion_scripted"]),