refactor(types): change module name (#4232)

* refactor(types): change module name

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

* chore(test): remove package import test

* chore: remove ruff exception

---------

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

tests/rewards/test_sarm_processor.py

@@ -25,7 +25,7 @@ import pandas as pd
 import pytest
 import torch
 
-from lerobot.types import TransitionKey
+from lerobot.lerobot_types import TransitionKey
 
 
 class MockDatasetMeta:

tests/rewards/test_topreward_processor.py

@@ -20,13 +20,13 @@ import pytest
 import torch
 
 from lerobot.configs import FeatureType, PipelineFeatureType, PolicyFeature
+from lerobot.lerobot_types import TransitionKey
 from lerobot.rewards.topreward.processor_topreward import (
     TOPREWARD_FEATURE_PREFIX,
     TOPREWARD_INPUT_KEYS,
     _expand_tasks,
     _prepare_video_batch,
 )
-from lerobot.types import TransitionKey
 from tests.utils import skip_if_package_missing
 
 # ---------------------------------------------------------------------------