Apply suggestions from code review

Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
Signed-off-by: Adil Zouitine <adilzouitinegm@gmail.com>

tests/envs/test_envs.py

@@ -22,8 +22,7 @@ from gymnasium.utils.env_checker import check_env
 
 import lerobot
 from lerobot.envs.factory import make_env, make_env_config
-from lerobot.processor.observation_processor import VanillaObservationProcessor
-from lerobot.processor.pipeline import RobotProcessor, TransitionIndex
+from lerobot.processor import RobotProcessor, TransitionIndex, VanillaObservationProcessor
 from tests.utils import require_env
 
 OBS_TYPES = ["state", "pixels", "pixels_agent_pos"]

tests/policies/test_policies.py

@@ -39,8 +39,7 @@ from lerobot.policies.factory import (
 )
 from lerobot.policies.normalize import Normalize, Unnormalize
 from lerobot.policies.pretrained import PreTrainedPolicy
-from lerobot.processor.observation_processor import VanillaObservationProcessor
-from lerobot.processor.pipeline import RobotProcessor, TransitionIndex
+from lerobot.processor import RobotProcessor, TransitionIndex, VanillaObservationProcessor
 from lerobot.utils.random_utils import seeded_context
 from tests.artifacts.policies.save_policy_to_safetensors import get_policy_stats
 from tests.utils import DEVICE, require_cpu, require_env, require_x86_64_kernel

tests/processor/test_observation_processor.py

@@ -18,7 +18,7 @@ import numpy as np
 import pytest
 import torch
 
-from lerobot.processor.observation_processor import (
+from lerobot.processor import (
     ImageProcessor,
     StateProcessor,
     VanillaObservationProcessor,

tests/processor/test_pipeline.py

@@ -25,7 +25,7 @@ import pytest
 import torch
 import torch.nn as nn
 
-from lerobot.processor.pipeline import EnvTransition, ProcessorStepRegistry, RobotProcessor
+from lerobot.processor import EnvTransition, ProcessorStepRegistry, RobotProcessor
 
 
 @dataclass

tests/processor/test_rename_processor.py

@@ -20,8 +20,7 @@ from pathlib import Path
 import numpy as np
 import torch
 
-from lerobot.processor.pipeline import ProcessorStepRegistry, RobotProcessor, TransitionIndex
-from lerobot.processor.rename_processor import RenameProcessor
+from lerobot.processor import ProcessorStepRegistry, RobotProcessor, TransitionIndex, RenameProcessor
 
 
 def test_basic_renaming():