admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-30 21:19:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(types): change module name

Browse Source 
Co-authored-by: saiteja6006 <saiteja6006@gmail.com>
This commit is contained in:
Steven Palma
2026-07-30 14:46:54 +02:00
parent 40a5e70352
commit fbe1474c9c
107 changed files with 150 additions and 106 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/groot/test_groot_lerobot.py
+1 -1
View File
@@ -27,11 +27,11 @@ import torch
pytest.importorskip("transformers", reason="groot requires the `groot` extra (transformers)")
from lerobot.lerobot_types import PolicyAction
from lerobot.policies.groot.configuration_groot import GrootConfig
from lerobot.policies.groot.modeling_groot import GrootPolicy
from lerobot.policies.groot.processor_groot import make_groot_pre_post_processors
from lerobot.processor import PolicyProcessorPipeline
from lerobot.types import PolicyAction
from lerobot.utils.device_utils import auto_select_torch_device
from tests.utils import require_cuda
tests/policies/groot/test_groot_n1_7.py
+1 -1
View File
@@ -27,6 +27,7 @@ from safetensors.torch import load_file
from torch import nn
from lerobot.configs import FeatureType, PolicyFeature
from lerobot.lerobot_types import TransitionKey
from lerobot.policies.factory import make_policy_config, make_pre_post_processors
from lerobot.policies.groot.configuration_groot import (
GROOT_ACTION_DECODE_TRANSFORM_LIBERO,
@@ -54,7 +55,6 @@ from lerobot.processor import (
PolicyProcessorPipeline,
RelativeActionsProcessorStep,
)
from lerobot.types import TransitionKey
from lerobot.utils.constants import ACTION, OBS_IMAGES, OBS_STATE
tests/policies/groot/test_groot_n1_7_oss_parity.py
+1 -1
View File
@@ -22,6 +22,7 @@ import numpy as np
import pytest
import torch
from lerobot.lerobot_types import TransitionKey
from lerobot.policies.groot.action_head.cross_attention_dit import AlternateVLDiT
from lerobot.policies.groot.groot_n1_7 import GR00TN17
from lerobot.policies.groot.processor_groot import (
@@ -30,7 +31,6 @@ from lerobot.policies.groot.processor_groot import (
GrootN17VLMEncodeStep,
_transform_n1_7_image_for_vlm_albumentations,
)
from lerobot.types import TransitionKey
from lerobot.utils.constants import OBS_STATE
OSS_REFERENCE_COMMIT = "ab88b50c718f6528e1df9dcbaf75865d1b604760"
tests/policies/groot/test_groot_state_dropout.py
+1 -1
View File
@@ -24,8 +24,8 @@ regularization never activated. These tests pin the train/eval split.
import torch
from lerobot.lerobot_types import TransitionKey
from lerobot.policies.groot.processor_groot import GrootN17PackInputsStep
from lerobot.types import TransitionKey
from lerobot.utils.constants import OBS_STATE