Cleanup policies imports

2025-03-08 11:08:40 +01:00
parent 6909b62a21
commit 80387285bb
15 changed files with 53 additions and 48 deletions
--- a/tests/test_control_robot.py
+++ b/tests/test_control_robot.py
@@ -41,9 +41,8 @@ from unittest.mock import patch

 import pytest

+from lerobot.common.policies import PreTrainedConfig, make_policy
 from lerobot.common.policies.act.configuration_act import ACTConfig
-from lerobot.common.policies.config import PreTrainedConfig
-from lerobot.common.policies.factory import make_policy
 from lerobot.common.robot_devices.control_configs import (
    CalibrateControlConfig,
    RecordControlConfig,