mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-15 08:39:49 +00:00
feat(dependencies): minimal default tag install (#3362)
This commit is contained in:
@@ -22,14 +22,12 @@ import torch
|
||||
|
||||
from lerobot.configs.types import FeatureType, NormalizationMode, PipelineFeatureType, PolicyFeature
|
||||
from lerobot.policies.smolvla.configuration_smolvla import SmolVLAConfig
|
||||
from lerobot.policies.smolvla.processor_smolvla import (
|
||||
SmolVLANewLineProcessor,
|
||||
make_smolvla_pre_post_processors,
|
||||
)
|
||||
from lerobot.policies.smolvla.processor_smolvla import make_smolvla_pre_post_processors
|
||||
from lerobot.processor import (
|
||||
AddBatchDimensionProcessorStep,
|
||||
DeviceProcessorStep,
|
||||
EnvTransition,
|
||||
NewLineTaskProcessorStep,
|
||||
NormalizerProcessorStep,
|
||||
ProcessorStep,
|
||||
RenameObservationsProcessorStep,
|
||||
@@ -108,7 +106,7 @@ def test_make_smolvla_processor_basic():
|
||||
assert len(preprocessor.steps) == 6
|
||||
assert isinstance(preprocessor.steps[0], RenameObservationsProcessorStep)
|
||||
assert isinstance(preprocessor.steps[1], AddBatchDimensionProcessorStep)
|
||||
assert isinstance(preprocessor.steps[2], SmolVLANewLineProcessor)
|
||||
assert isinstance(preprocessor.steps[2], NewLineTaskProcessorStep)
|
||||
# Step 3 would be TokenizerProcessorStep but it's mocked
|
||||
assert isinstance(preprocessor.steps[4], DeviceProcessorStep)
|
||||
assert isinstance(preprocessor.steps[5], NormalizerProcessorStep)
|
||||
@@ -120,8 +118,8 @@ def test_make_smolvla_processor_basic():
|
||||
|
||||
|
||||
def test_smolvla_newline_processor_single_task():
|
||||
"""Test SmolVLANewLineProcessor with single task string."""
|
||||
processor = SmolVLANewLineProcessor()
|
||||
"""Test NewLineTaskProcessorStep with single task string."""
|
||||
processor = NewLineTaskProcessorStep()
|
||||
|
||||
# Test with task that doesn't have newline
|
||||
transition = create_transition(complementary_data={"task": "test task"})
|
||||
@@ -135,8 +133,8 @@ def test_smolvla_newline_processor_single_task():
|
||||
|
||||
|
||||
def test_smolvla_newline_processor_list_of_tasks():
|
||||
"""Test SmolVLANewLineProcessor with list of task strings."""
|
||||
processor = SmolVLANewLineProcessor()
|
||||
"""Test NewLineTaskProcessorStep with list of task strings."""
|
||||
processor = NewLineTaskProcessorStep()
|
||||
|
||||
# Test with list of tasks
|
||||
tasks = ["task1", "task2\n", "task3"]
|
||||
@@ -147,8 +145,8 @@ def test_smolvla_newline_processor_list_of_tasks():
|
||||
|
||||
|
||||
def test_smolvla_newline_processor_empty_transition():
|
||||
"""Test SmolVLANewLineProcessor with empty transition."""
|
||||
processor = SmolVLANewLineProcessor()
|
||||
"""Test NewLineTaskProcessorStep with empty transition."""
|
||||
processor = NewLineTaskProcessorStep()
|
||||
|
||||
# Test with no complementary_data
|
||||
transition = create_transition()
|
||||
@@ -361,8 +359,8 @@ def test_smolvla_processor_without_stats():
|
||||
|
||||
|
||||
def test_smolvla_newline_processor_state_dict():
|
||||
"""Test SmolVLANewLineProcessor state dict methods."""
|
||||
processor = SmolVLANewLineProcessor()
|
||||
"""Test NewLineTaskProcessorStep state dict methods."""
|
||||
processor = NewLineTaskProcessorStep()
|
||||
|
||||
# Test state_dict (should be empty)
|
||||
state = processor.state_dict()
|
||||
@@ -380,8 +378,8 @@ def test_smolvla_newline_processor_state_dict():
|
||||
|
||||
|
||||
def test_smolvla_newline_processor_transform_features():
|
||||
"""Test SmolVLANewLineProcessor transform_features method."""
|
||||
processor = SmolVLANewLineProcessor()
|
||||
"""Test NewLineTaskProcessorStep transform_features method."""
|
||||
processor = NewLineTaskProcessorStep()
|
||||
|
||||
# Test transform_features
|
||||
features = {
|
||||
|
||||
Reference in New Issue
Block a user