27 lines
588 B
Python
27 lines
588 B
Python
|
"""LeRobot backend registration."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import sys
|
||
|
|
|
||
|
|
try:
|
||
|
|
from wall_x.data._registry import register_module
|
||
|
|
from wall_x.data.backends.lerobot.build import (
|
||
|
|
build,
|
||
|
|
load_trainer_data_config,
|
||
|
|
load_trainer_data_config_from_yaml_dict,
|
||
|
|
)
|
||
|
|
|
||
|
|
register_module("lerobot", sys.modules[__name__])
|
||
|
|
except ImportError as _e:
|
||
|
|
from wall_x.data._registry import record_import_error
|
||
|
|
|
||
|
|
record_import_error("lerobot", _e)
|
||
|
|
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
"build",
|
||
|
|
"load_trainer_data_config",
|
||
|
|
"load_trainer_data_config_from_yaml_dict",
|
||
|
|
]
|