From 8366648a8a4726e9223dafbb3d6a9d6c27e021c7 Mon Sep 17 00:00:00 2001
From: Remi Cadene <re.cadene@gmail.com>
Date: Fri, 31 May 2024 10:15:46 +0000
Subject: [PATCH] revert

---
 lerobot/__init__.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lerobot/__init__.py b/lerobot/__init__.py
index 8e554998..a5a90fb4 100644
--- a/lerobot/__init__.py
+++ b/lerobot/__init__.py
@@ -138,8 +138,8 @@ available_policies = [
 
 # keys and values refer to yaml files
 available_policies_per_env = {
-    "aloha": ["act", "diffusion"],
-    "pusht": ["act", "diffusion"],
+    "aloha": ["act"],
+    "pusht": ["diffusion"],
     "xarm": ["tdmpc"],
     "dora_aloha_real": ["act_real"],
 }