working changes

This commit is contained in:
jade.choghari@huggingface.co
2025-11-18 13:55:45 +01:00
parent 4c67330430
commit e5efb6b6dc
8 changed files with 159 additions and 13 deletions
+13
View File
@@ -1,4 +1,7 @@
from lerobot.envs.factory import make_env, make_env_config
from lerobot.envs.utils import add_envs_task, preprocess_observation
from lerobot.processor.pipeline import PolicyProcessorPipeline
from lerobot.processor.observation_processor import LiberoProcessorStep
config = make_env_config("libero", task="libero_spatial")
envs_dict = make_env(config)
env = envs_dict["libero_spatial"][0]
@@ -7,3 +10,13 @@ seed = 42
# First rollout
obs1, info1 = env.reset(seed=seed)
observation = preprocess_observation(obs1)
observation = add_envs_task(env, observation)
libero_preprocessor = PolicyProcessorPipeline(
steps=[
LiberoProcessorStep(),
]
)
observation = libero_preprocessor(observation)