refactor(types): change module name

Co-authored-by: saiteja6006 <saiteja6006@gmail.com>

tests/processor/test_batch_conversion.py

@@ -16,9 +16,9 @@
 
 import torch
 
+from lerobot.lerobot_types import TransitionKey
 from lerobot.processor import DataProcessorPipeline
 from lerobot.processor.converters import batch_to_transition, transition_to_batch
-from lerobot.types import TransitionKey
 from lerobot.utils.constants import ACTION, DONE, OBS_IMAGE, OBS_PREFIX, OBS_STATE, REWARD, TRUNCATED
 
 

tests/processor/test_converters.py

@@ -18,13 +18,13 @@ import numpy as np
 import pytest
 import torch
 
+from lerobot.lerobot_types import TransitionKey
 from lerobot.processor.converters import (
     batch_to_transition,
     create_transition,
     to_tensor,
     transition_to_batch,
 )
-from lerobot.types import TransitionKey
 from lerobot.utils.constants import ACTION, DONE, OBS_STATE, OBS_STR, REWARD
 
 

tests/processor/test_device_processor.py

@@ -19,9 +19,9 @@ import pytest
 import torch
 
 from lerobot.configs.types import FeatureType, PipelineFeatureType, PolicyFeature
+from lerobot.lerobot_types import TransitionKey
 from lerobot.processor import DataProcessorPipeline, DeviceProcessorStep
 from lerobot.processor.converters import create_transition, identity_transition
-from lerobot.types import TransitionKey
 from lerobot.utils.constants import ACTION, OBS_IMAGE, OBS_STATE
 
 

tests/processor/test_observation_processor.py

@@ -19,9 +19,9 @@ import pytest
 import torch
 
 from lerobot.configs.types import FeatureType, PipelineFeatureType
+from lerobot.lerobot_types import TransitionKey
 from lerobot.processor import VanillaObservationProcessorStep
 from lerobot.processor.converters import create_transition
-from lerobot.types import TransitionKey
 from lerobot.utils.constants import OBS_ENV_STATE, OBS_IMAGE, OBS_IMAGES, OBS_STATE
 from tests.conftest import assert_contract_is_typed
 

tests/processor/test_pipeline_from_pretrained_helpers.py

@@ -30,13 +30,13 @@ import torch
 from safetensors.torch import save_file
 
 from lerobot.configs import PipelineFeatureType, PolicyFeature
+from lerobot.lerobot_types import EnvTransition
 from lerobot.processor.pipeline import (
     DataProcessorPipeline,
     ProcessorMigrationError,
     ProcessorStep,
     ProcessorStepRegistry,
 )
-from lerobot.types import EnvTransition
 
 # Simplified Config Loading Tests
 

tests/processor/test_render_messages_processor.py

@@ -7,9 +7,9 @@ pytest.importorskip("datasets", reason="datasets is required (install lerobot[da
 import torch  # noqa: E402
 
 from lerobot.configs.recipe import MessageTurn, TrainingRecipe  # noqa: E402
+from lerobot.lerobot_types import TransitionKey  # noqa: E402
 from lerobot.processor.converters import create_transition  # noqa: E402
 from lerobot.processor.render_messages_processor import RenderMessagesStep  # noqa: E402
-from lerobot.types import TransitionKey  # noqa: E402
 
 
 def test_render_messages_step_noops_without_language_columns():

tests/processor/test_tokenizer_processor.py

@@ -25,9 +25,9 @@ import pytest
 import torch
 
 from lerobot.configs.types import FeatureType, PipelineFeatureType, PolicyFeature
+from lerobot.lerobot_types import TransitionKey
 from lerobot.processor import DataProcessorPipeline, TokenizerProcessorStep
 from lerobot.processor.converters import create_transition, identity_transition
-from lerobot.types import TransitionKey
 from lerobot.utils.constants import (
     ACTION,
     OBS_IMAGE,