mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-24 02:06:15 +00:00
removing useless pre-processor
This commit is contained in:
@@ -21,7 +21,6 @@ import torch
|
|||||||
from lerobot.policies.vla_jepa.configuration_vla_jepa import VLAJEPAConfig
|
from lerobot.policies.vla_jepa.configuration_vla_jepa import VLAJEPAConfig
|
||||||
from lerobot.processor import (
|
from lerobot.processor import (
|
||||||
AddBatchDimensionProcessorStep,
|
AddBatchDimensionProcessorStep,
|
||||||
ComplementaryDataProcessorStep,
|
|
||||||
DeviceProcessorStep,
|
DeviceProcessorStep,
|
||||||
EnvTransition,
|
EnvTransition,
|
||||||
NormalizerProcessorStep,
|
NormalizerProcessorStep,
|
||||||
@@ -142,12 +141,3 @@ def make_vla_jepa_pre_post_processors(
|
|||||||
to_output=transition_to_policy_action,
|
to_output=transition_to_policy_action,
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
@ProcessorStepRegistry.register(name="vla_jepa_new_line_processor")
|
|
||||||
class VLAJEPANewLineProcessor(ComplementaryDataProcessorStep):
|
|
||||||
def complementary_data(self, complementary_data):
|
|
||||||
return complementary_data
|
|
||||||
|
|
||||||
def transform_features(self, features):
|
|
||||||
return features
|
|
||||||
|
|||||||
Reference in New Issue
Block a user