mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-14 05:22:14 +00:00
simplify molmoact2 package imports
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user