admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-15 16:49:55 +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.
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
from lerobot.envs.utils import preprocess_observation
from lerobot.processor.observation_processor import LiberoProcessorStep
from lerobot.processor.pipeline import PolicyProcessorPipeline
seed = 42
np.random.seed(seed)
B = 5
obs1 = {
"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),
},
"robot_state": {
@@ -42,8 +44,8 @@ obs1 = {
"joints": {
"pos": np.random.randn(B, 7),
"vel": np.random.randn(B, 7),
}
}
},
},
}
observation = preprocess_observation(obs1)