Move make_policy_config

This commit is contained in:
Simon Alibert
2025-03-08 12:42:43 +01:00
parent 00698305a3
commit 57e0bdb491
5 changed files with 11 additions and 32 deletions

View File

@@ -40,12 +40,11 @@ from lerobot.common.datasets.utils import (
unflatten_dict,
)
from lerobot.common.envs.factory import make_env_config
from lerobot.common.policies.factory import make_policy_config
from lerobot.common.robot_devices.robots.utils import make_robot
from lerobot.configs.default import DatasetConfig
from lerobot.configs.train import TrainPipelineConfig
from tests.fixtures.constants import DUMMY_CHW, DUMMY_HWC, DUMMY_REPO_ID
from tests.utils import require_x86_64_kernel
from tests.utils import make_policy_config, require_x86_64_kernel
@pytest.fixture