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

- 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
+2 -2
View File
@@ -18,9 +18,9 @@ from __future__ import annotations
import logging
from typing import TypedDict
from typing_extensions import Unpack
from torch import nn
from typing_extensions import Unpack
from lerobot.configs.policies import PreTrainedConfig
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.tdmpc.configuration_tdmpc import TDMPCConfig
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: