mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-22 20:19:43 +00:00
add back previously removed imports
This commit is contained in:
@@ -16,6 +16,9 @@
|
|||||||
|
|
||||||
from dataclasses import dataclass, field
|
from dataclasses import dataclass, field
|
||||||
|
|
||||||
|
from lerobot import (
|
||||||
|
policies, # noqa: F401
|
||||||
|
)
|
||||||
from lerobot.datasets.transforms import ImageTransformsConfig
|
from lerobot.datasets.transforms import ImageTransformsConfig
|
||||||
from lerobot.datasets.video_utils import get_safe_default_codec
|
from lerobot.datasets.video_utils import get_safe_default_codec
|
||||||
|
|
||||||
|
|||||||
@@ -26,6 +26,14 @@ from lerobot.constants import OBS_IMAGES, OBS_STATE
|
|||||||
from lerobot.datasets.utils import build_dataset_frame, hw_to_dataset_features
|
from lerobot.datasets.utils import build_dataset_frame, hw_to_dataset_features
|
||||||
|
|
||||||
# NOTE: Configs need to be loaded for the client to be able to instantiate the policy config
|
# NOTE: Configs need to be loaded for the client to be able to instantiate the policy config
|
||||||
|
from lerobot.policies import ( # noqa: F401
|
||||||
|
ACTConfig,
|
||||||
|
DiffusionConfig,
|
||||||
|
PI0Config,
|
||||||
|
PI05Config,
|
||||||
|
SmolVLAConfig,
|
||||||
|
VQBeTConfig,
|
||||||
|
)
|
||||||
from lerobot.robots.robot import Robot
|
from lerobot.robots.robot import Robot
|
||||||
from lerobot.utils.utils import init_logging
|
from lerobot.utils.utils import init_logging
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user