add back previously removed imports

This commit is contained in:
Pepijn
2025-09-24 17:26:50 +02:00
parent 0fd1b70d03
commit f53fa00952
2 changed files with 11 additions and 0 deletions
+3
View File
@@ -16,6 +16,9 @@
from dataclasses import dataclass, field
from lerobot import (
policies, # noqa: F401
)
from lerobot.datasets.transforms import ImageTransformsConfig
from lerobot.datasets.video_utils import get_safe_default_codec
+8
View File
@@ -26,6 +26,14 @@ from lerobot.constants import OBS_IMAGES, OBS_STATE
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
from lerobot.policies import ( # noqa: F401
ACTConfig,
DiffusionConfig,
PI0Config,
PI05Config,
SmolVLAConfig,
VQBeTConfig,
)
from lerobot.robots.robot import Robot
from lerobot.utils.utils import init_logging