admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-22 03:59:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(factory): Clean up imports in factory.py

Browse Source 
- Removed unused import of IdentityProcessor to streamline the code.
This commit is contained in:
Adil Zouitine
2025-07-24 17:25:57 +02:00
committed by Steven Palma
parent 023b8f3466
commit 9b1138171e
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/factory.py
+2 -2
View File
@@ -18,9 +18,9 @@ from __future__ import annotations
import logging import logging
from typing import TypedDict from typing import TypedDict
from typing_extensions import Unpack
from torch import nn from torch import nn
from typing_extensions import Unpack
from lerobot.configs.policies import PreTrainedConfig from lerobot.configs.policies import PreTrainedConfig
from lerobot.configs.types import FeatureType from lerobot.configs.types import FeatureType
@@ -38,7 +38,7 @@ from lerobot.policies.sac.reward_model.configuration_classifier import RewardCla
from lerobot.policies.smolvla.configuration_smolvla import SmolVLAConfig from lerobot.policies.smolvla.configuration_smolvla import SmolVLAConfig
from lerobot.policies.tdmpc.configuration_tdmpc import TDMPCConfig from lerobot.policies.tdmpc.configuration_tdmpc import TDMPCConfig
from lerobot.policies.vqbet.configuration_vqbet import VQBeTConfig from lerobot.policies.vqbet.configuration_vqbet import VQBeTConfig
from lerobot.processor.pipeline import IdentityProcessor, RobotProcessor from lerobot.processor.pipeline import RobotProcessor
def get_policy_class(name: str) -> PreTrainedPolicy: def get_policy_class(name: str) -> PreTrainedPolicy: