admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-08-08 17:39:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(g05): consolidate policy modules

Browse Source
This commit is contained in:
Pepijn
2026-07-30 13:00:40 +02:00
parent 65a8b40484
commit 12e7416473
7 changed files with 2200 additions and 2284 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/g05/test_action_codec_g05.py
+1 -1
View File
@@ -16,7 +16,7 @@
import torch
from lerobot.policies.g05.action_codec_g05 import G05NativeActionCodec, _BinarySequenceCodec
from lerobot.policies.g05.modeling_g05 import G05NativeActionCodec, _BinarySequenceCodec
def _tiny_codec_config() -> dict:
tests/policies/g05/test_g05.py
+1 -2
View File
@@ -28,8 +28,7 @@ from lerobot.configs.policies import PreTrainedConfig
from lerobot.configs.types import FeatureType, PolicyFeature
from lerobot.policies.factory import get_policy_class, make_policy_config, make_pre_post_processors
from lerobot.policies.g05.configuration_g05 import G05_CAMERA_PROFILES, G05_EMBODIMENT_MAPPINGS, G05Config
from lerobot.policies.g05.modeling_g05 import G05Policy
from lerobot.policies.g05.native_g05 import G05_RUNTIME_PREDICT_COT, G05NativeBackend
from lerobot.policies.g05.modeling_g05 import G05_RUNTIME_PREDICT_COT, G05NativeBackend, G05Policy
from lerobot.processor import PolicyProcessorPipeline
from lerobot.utils.constants import ACTION, OBS_STATE, POLICY_PREPROCESSOR_DEFAULT_NAME