admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-22 03:59:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix style

Browse Source
This commit is contained in:
jade.choghari@huggingface.co
2025-11-18 14:25:48 +01:00
parent b4b5d057b1
commit e91e48b79c
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/processor/test_libero_processor.py
+8 -6
View File
@@ -15,17 +15,19 @@
# limitations under the License. # limitations under the License.
import numpy as np import numpy as np
from lerobot.envs.utils import preprocess_observation
from lerobot.processor.pipeline import PolicyProcessorPipeline
from lerobot.processor.observation_processor import LiberoProcessorStep
import torch import torch
from lerobot.envs.utils import preprocess_observation
from lerobot.processor.observation_processor import LiberoProcessorStep
from lerobot.processor.pipeline import PolicyProcessorPipeline
seed = 42 seed = 42
np.random.seed(seed) np.random.seed(seed)
B = 5 B = 5
obs1 = { obs1 = {
"pixels": { "pixels": {
"image": (np.random.rand(B, 256, 256, 3) * 255).astype(np.uint8), "image": (np.random.rand(B, 256, 256, 3) * 255).astype(np.uint8),
"image2": (np.random.rand(B, 256, 256, 3) * 255).astype(np.uint8), "image2": (np.random.rand(B, 256, 256, 3) * 255).astype(np.uint8),
}, },
"robot_state": { "robot_state": {
@@ -42,8 +44,8 @@ obs1 = {
"joints": { "joints": {
"pos": np.random.randn(B, 7), "pos": np.random.randn(B, 7),
"vel": np.random.randn(B, 7), "vel": np.random.randn(B, 7),
} },
} },
} }
observation = preprocess_observation(obs1) observation = preprocess_observation(obs1)