From b858ba1b6ce4f8be1849c24a61f5d54c917f45a3 Mon Sep 17 00:00:00 2001
From: hq-fang <71356829+hq-fang@users.noreply.github.com>
Date: Tue, 19 May 2026 21:57:45 +0000
Subject: [PATCH] simplify molmoact2 package imports

---
 src/lerobot/policies/molmoact2/__init__.py | 14 ++------------
 1 file changed, 2 insertions(+), 12 deletions(-)

diff --git a/src/lerobot/policies/molmoact2/__init__.py b/src/lerobot/policies/molmoact2/__init__.py
index dfaea261b..bfef53bb2 100644
--- a/src/lerobot/policies/molmoact2/__init__.py
+++ b/src/lerobot/policies/molmoact2/__init__.py
@@ -15,17 +15,7 @@
 # limitations under the License.
 
 from .configuration_molmoact2 import MolmoAct2Config
+from .modeling_molmoact2 import MolmoAct2Policy
+from .processor_molmoact2 import make_molmoact2_pre_post_processors
 
 __all__ = ["MolmoAct2Config", "MolmoAct2Policy", "make_molmoact2_pre_post_processors"]
-
-
-def __getattr__(name):
-    if name == "MolmoAct2Policy":
-        from .modeling_molmoact2 import MolmoAct2Policy
-
-        return MolmoAct2Policy
-    if name == "make_molmoact2_pre_post_processors":
-        from .processor_molmoact2 import make_molmoact2_pre_post_processors
-
-        return make_molmoact2_pre_post_processors
-    raise AttributeError(name)