tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Cleanup policies imports

Browse Source
This commit is contained in:
Simon Alibert
2025-03-08 11:08:40 +01:00
parent 6909b62a21
commit 80387285bb
15 changed files with 53 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/scripts/save_policy_to_safetensors.py
View File

@@ -21,7 +21,8 @@ from safetensors.torch import save_file
from lerobot.common.datasets.factory import make_dataset
from lerobot.common.optim.factory import make_optimizer_and_scheduler
from lerobot.common.policies.factory import make_policy, make_policy_config
from lerobot.common.policies import make_policy
from lerobot.common.policies.factory import make_policy_config
from lerobot.common.utils.random_utils import set_seed
from lerobot.configs.default import DatasetConfig
from lerobot.configs.train import TrainPipelineConfig

3
tests/test_control_robot.py
View File

@@ -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,