admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-25 10:46:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

simplify molmoact2 package imports

Browse Source
This commit is contained in:
hq-fang
2026-05-19 21:57:45 +00:00
parent e870af119f
commit b858ba1b6c
1 changed files with 2 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/molmoact2/__init__.py
+2 -12
View File
@@ -15,17 +15,7 @@
# limitations under the License. # limitations under the License.
from .configuration_molmoact2 import MolmoAct2Config 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"] __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)